LeetCode 1356. 根据数字二进制下 1 的数目排序

原题目：https://leetcode-cn.com/problems/sort-integers-by-the-number-of-1-bits/

 

思路：

计算出每一个数字的1的位数。重写排序规则

 

代码：

class Solution {
    #define N 10010
    int bit[N];
public:
    int get(int x){
        int res=0;
        while (x) res+=x&1,x>>=1;
        return res;
    }
    vector<int> sortByBits(vector<int>& arr) {
        for (auto x:arr) bit[x]=get(x);
        sort(arr.begin(),arr.end(),[&](int x,int y){
            return bit[x]==bit[y]?x<y:bit[x]<bit[y];
        });
        return arr;
    }
};