每日一题 2020.11.6

1356. 根据数字二进制下 1 的数目排序

给你一个整数数组 arr 。请你将数组中的元素按照其二进制表示中数字 1 的数目升序排序。

如果存在多个数字二进制中 1 的数目相同，则必须将它们按照数值大小升序排列。

请你返回排序后的数组。

思路：预处理出各数的含1数量，然后进行排序

var nums = [1e4+10]int{}

//初始化函数 自动调用
func init(){
    for i:=1;i<=1e4;i++{
        nums[i] = nums[i>>1] + i&1
    }
}

func sortByBits(arr []int) []int {
    sort.Slice(arr ,func(i,j int)bool{
        if nums[arr[i]] == nums[arr[j]]{
            return arr[i]<arr[j]
        }
        return nums[arr[i]] < nums[arr[j]]
    })
    return arr
}