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

hn小菜鸡2025-06-09 19:48

题目：

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

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

请你返回排序后的数组。

提示：

1 <= arr.length <= 500
0 <= arr[i] <= 10^4

思路：首先，根据 1的个数和当前数值，生成一个新的数字，来存储每一个数字的 1的个数和本身的值。接下来，将存储的数字，还原成最初始的数字，并根据 1的个数和当前数值排序。

代码：

java 复制代码

class Solution {
    public int[] sortByBits(int[] arr) {
        int length = arr.length;
        for (int i = 0; i < length; i++) {
        // 因为arr[i]最大是10^4，还原时 arr[i]是余数
        // 因此 %= 10^5 放大也是放大10^5
            arr[i] = Integer.bitCount(arr[i]) * 100000 + arr[i];
        }
        Arrays.sort(arr);
        for (int i = 0; i < length; i++) {
            arr[i] %= 100000;
        }
        return arr;
    }
}

性能：

时间复杂度o（nlogn）快排时间

空间复杂度o（logn）快排递归调用栈