南邮 OJ 1024 01排序

01排序

时间限制(普通/Java) : 1000 MS/ 3000 MS          运行内存限制 : 65536 KByte
总提交 : 935            测试通过 : 351 

比赛描述

将01串首先按长度排序，长度相同时，按1的个数多少进行排序，1的个数相同时再按ASCII码值排序。

输入

输入数据中含有一些01串，01串的长度不大于256个字符。

输出

重新排列01串的顺序。使得串按基本描述的方式排序。

样例输入

10011111

00001101

1010101

1

0

1100

样例输出

0

1

1100

1010101

00001101

10011111

题目来源

ZJUT

#include <iostream>
#include <string>
#include <algorithm>
#define MAXNUM 256
using namespace std;

bool compareBy1s(string str0,string str1){
if(count(str0.begin(),str0.end(),'1')<count(str1.begin(),str1.end(),'1')){
return 1;
}else if(count(str0.begin(),str0.end(),'1')>count(str1.begin(),str1.end(),'1')){
return 0;
}else{
return str0<str1;
}
}
bool compare(string str0,string str1){
if(str0.length()<str1.length()){
return 1;
}else if(str0.length()>str1.length()){
return 0;
}else{
return compareBy1s(str0,str1);
}
}
int main(void){
string str[MAXNUM];
int i=0,n=0;
while(cin>>str[n++]);
--n;
stable_sort(str,str+n,compare);
for(i=0;i<n;++i){
cout<<str[i]<<endl;
}
}