按1的个数排序

Description:

有一些01字串，将其按1的个数的多少的顺序进行输出。

 

Sample Input:

10011111
00001101
1010101
1
0
1100

Sample Output:

0
1
1100
00001101
1010101
10011111

 

#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
using namespace std;

bool cmp(const string &a, const string &b){
    int n1 = count(a.begin(), a.end(), '1');
    int n2 = count(b.begin(), b.end(), '1');
    //彻底修改排序规则，只按1的数量排序，
    //如果1的数量相等，则按出现的先后顺序排序
    //否则会按ASCII码大小排序
    //只能用“>”，或“<”，不能用“=”
    return n1 != n2 ? n1 < n2 : n1 < n2;
}

int main(){
    string s;
    vector<string> v;
    while(cin >> s){
        v.push_back(s);
    }
    sort(v.begin(), v.end(), cmp);
    for(vector<string>::iterator it = v.begin(); it != v.end(); it++){
        cout << *it << endl;
    }
    system("pause");
    return 0;
}