1. 程式人生 > 實用技巧 >力扣每日一題:根據數字二進位制下 1 的數目排序

力扣每日一題:根據數字二進位制下 1 的數目排序

給你一個整數陣列arr。請你將陣列中的元素按照其二進位制表示中數字 1 的數目升序排序。

如果存在多個數字二進位制中1的數目相同,則必須將它們按照數值大小升序排列。

請你返回排序後的陣列。

示例 1:

輸入:arr = [0,1,2,3,4,5,6,7,8]
輸出:[0,1,2,4,8,3,5,6,7]
解釋:[0] 是唯一一個有 0 個 1 的數。
[1,2,4,8] 都有 1 個 1 。
[3,5,6] 有 2 個 1 。
[7] 有 3 個 1 。
按照 1 的個數排序得到的結果陣列為 [0,1,2,4,8,3,5,6,7]
示例 2:

輸入:arr = [1024,512,256,128,64,32,16,8,4,2,1]
輸出:[1,2,4,8,16,32,64,128,256,512,1024]
解釋:陣列中所有整數二進位制下都只有 1 個 1 ,所以你需要按照數值大小將它們排序。
示例 3:

輸入:arr = [10000,10000]
輸出:[10000,10000]
示例 4:

輸入:arr = [2,3,5,7,11,13,17,19]
輸出:[2,3,5,17,7,11,13,19]
示例 5:

輸入:arr = [10,100,1000,10000]
輸出:[10,100,10000,1000]

提示:

1 <= arr.length <= 500
0 <= arr[i] <= 10^4

一道簡單的位運算 + 排序題。
1.此處不能用map<int,int>來儲存二進位制中1的個數以及數字本身。因為map中first是唯一的,當first相同的數字進入map後,會刪除之後插入的,只剩下第一個插入的{first,second}。
2.因為sort函式對pair排序是預設對first先排序,然後對second後排序,所以用vector<pair<int,int> > ans來儲存二進位制中1的個數以及數字本身。

AC程式碼:

class Solution {
public:
    vector<int> sortByBits(vector<int>& arr) {
        vector<pair<int ,int> > ans;
        vector<int> res;
        for(int i = 0 ; i < arr.size() ; i ++){
            int sum = 0;
            for(int j = 31 ; ~j ; j --){
                int x = (arr[i] >> j) & 1;
                if(x == 1) sum ++;
            }
            ans.push_back({sum,arr[i]});    
        }
        sort(ans.begin() , ans.end());
        for(auto it : ans) res.push_back(it.second);
        return res;
    }
};