力扣每日一題:根據數字二進位制下 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; } };