EOJ 3036 按数据中1的位数排序（位运算+排序）

题目

http://acm.ecnu.edu.cn/problem/3036/

题意：输入n个数，按照 64 位二进制补码表示中 1 的位数从大到小进行排序。若两个数的二进制表示中 1 的位数相同，则按照数本身值由小到大排序。

解题思路

实质是求二进制表示中1的个数。

利用位运算来实现，每次将n和n-1做与运算，即把n二进制表示中最右边的1去掉。重复操作直到n变为0，操作次数即二进制中1的个数。

AC代码

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;

const int maxn = 10005;
LL num[maxn];

int getOneCount(LL a) //二进制中1的个数(正负数都适用)
{
    int cnt = 0;
    while (a != 0)
    {
        cnt++;
        a &= (a-1); //去掉最右边的1
    }
    return cnt;
}

bool cmp(LL a, LL b) //按二进制中1的个数降序，相同时按值升序
{
    int one_a = getOneCount(a), one_b = getOneCount(b);
    return (one_a == one_b) ? (a < b) : (one_a > one_b);
}

int main()
{
    int T, n;
    cin >> T;
    for (int kase = 0; kase < T; ++kase)
    {
        cin >> n;
        for(int i = 0; i<n; ++i)
            cin >> num[i];
        sort(num, num+n, cmp); //自定义排序
        cout << "case #" << kase << ':' << endl;
        for(int i = 0; i < n-1; ++i)
            cout << num[i] << ' ';
        cout << num[n-1] << endl;
    }
    return 0;
}