LeetCode1356.根据数字二级制下1的数目排序

解法一

1. 统计出数组中各数的二进制下1的个数

public int countOf1(int n){
        int count = 0;
        while(n != 0){
            count += n & 1;
            n = n >>> 1; 
        }
        return count;
    }

2. 根据各数字二进制下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;
        }
    }

3. 完整代码

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;    
    }