直接除基取余法暴力
class Solution {
int m[10005];
public:
vector<int> sortByBits(vector<int>& arr) {
memset(m, 0, sizeof(m));
for(int i : arr)
{
int x = i; // 不改变原数组
if(!m[x]) // 未计算过的数字
{
while(x) // 除基取余法
{
if(x % 2) ++m[i];
x /= 2;
}
}
}
sort(arr.begin(), arr.end(),
[&](const int a, const int b){return m[a] == m[b] ? a < b : m[a] < m[b];});
return arr;
}
};
用时16ms
分析答案中采用的位运算:
int get(int x){
int res=0;
while (x) res+=x&1,x>>=1;
return res;
}
x & 1即判断x对应的二进制末位是否为1,res+过后,x >>= 1整体右移一位,最后一位被移掉,达到统计对应二进制中1的个数的效果,更快
修改后用时12ms
#include <bits/stdc++.h>
using namespace std;
class Solution {
int m[10005];
public:
vector<int> sortByBits(vector<int>& arr) {
memset(m, 0, sizeof(m));
for(int i : arr)
{
int x = i; // 不改变原数组
if(!m[x]) // 未计算过的数字
{
while(x) // 除基取余法
{
if(x & 1) ++m[i];
x >>= 1;
}
}
}
sort(arr.begin(), arr.end(),
[&](const int a, const int b){return m[a] == m[b] ? a < b : m[a] < m[b];});
return arr;
}
};