解法一
- 统计出数组中各数的二进制下1的个数
public int countOf1(int n){
int count = 0;
while(n != 0){
count += n & 1;
n = n >>> 1;
}
return count;
}
- 根据各数字二进制下1的数目排序,在这里刚好复习下插入排序
public void insertSort(int[][] arr){
for(int i = 1; i < arr.length; i++){
int j;
int[] temp = arr[i];
for( j = i - 1; j >= 0; j--){
if(arr[j][1] > temp[1] || (arr[j][1] == temp[1] && arr[j][0] > temp[0])){
arr[j+1] = arr[j];
} else {
break;
}
}
arr[j+1] = temp;
}
}
- 完整代码
class Solution {
public int[] sortByBits(int[] arr) {
int[] res = new int[arr.length];
int[][] count = new int[arr.length][2];
//插入排序
for(int i = 0; i < arr.length; i++){
count[i][0] = arr[i];
count[i][1] = countOf1(arr[i]);
}
insertSort(count);
for(int i = 0 ; i < res.length; i++){
res[i] = count[i][0];
}
return res;
}
public void insertSort(int[][] arr){
for(int i = 1; i < arr.length; i++){
int j;
int[] temp = arr[i];
for( j = i - 1; j >= 0; j--){
if(arr[j][1] > temp[1] || (arr[j][1] == temp[1] && arr[j][0] > temp[0])){
arr[j+1] = arr[j];
} else {
break;
}
}
arr[j+1] = temp;
}
}
public int countOf1(int n){
int count = 0;
while(n != 0){
count += n & 1;
n = n >>> 1;
}
return count;
}
}
解法二
用Java的bitCount
方法,bitCount
方法可以统计出数在二进制下1的个数。该方法时间复杂度O(N)
public int[] sortByBits(int[] arr){
int[] res = new int[arr.length];
for(int i = 0; i < arr.length; i++)
res[i] = Integer.bitCount(arr[i]) * 100000 + arr[i];
Arrays.sort(res);
for(int i = 0; i < res.length; i++)
res[i] = res[i] % 100000;
return res;
}