01串排序(stl set使用)

 

Description:

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

 

Input:

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

Output:

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

Sample Input:

10011111
00001101
1010101
1
0
1100

Sample Output:

0
1
1100
1010101
00001101
10011111

#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <set>
#include <iterator>

using namespace std;

class cmp
{
public:
    bool operator()(const string &str1, const string &str2)
    {
        if(str1.length() != str2.length())
            return str1.length() < str2.length();
        
        int c1 = count(str1.begin(), str1.end(), '1');
        int c2 = count(str2.begin(), str2.end(), '1');
        return (c1 != c2) ? c1 < c2 : str1 < str2;
    }
};

int main(int argc, char *argv[])
{
    //freopen("input.txt", "rt", stdin);
    //freopen("output.txt", "wt", stdout);
    
    multiset<string, cmp> set;
    string str;
    
    while(cin >> str)
        set.insert(str);
    
    ostream_iterator<string> os(cout, "/n");
    copy(set.begin(), set.end(), os);
    cout << endl;
    
    return EXIT_SUCCESS;
}