题目地址:https://leetcode-cn.com/problems/sort-integers-by-the-number-of-1-bits/
给你一个整数数组arr。 请你将数组中的元素按照其二进制表示中数字 1 的数目升序排序。 如果存在多个数字二进制中1的数目相同,则必须将它们按照数值大小升序排列。 请你返回排序后的数组。
示例 1:
输入:arr = [0,1,2,3,4,5,6,7,8]
输出:[0,1,2,4,8,3,5,6,7]
解释:[0] 是唯一一个有 0 个 1 的数。[1,2,4,8] 都有 1 个 1 。[3,5,6] 有 2 个 1 。[7] 有 3 个 1 。按照 1 的个数排序得到的结果数组为 [0,1,2,4,8,3,5,6,7]
示例 2:
输入:arr = [1024,512,256,128,64,32,16,8,4,2,1]
输出:[1,2,4,8,16,32,64,128,256,512,1024]
解释:数组中所有整数二进制下都只有 1 个 1 ,所以你需要按照数值大小将它们排序。
示例 3:
输入:arr = [10000,10000]
输出:[10000,10000]
示例 4:
输入:arr = [2,3,5,7,11,13,17,19]
输出:[2,3,5,17,7,11,13,19]
示例 5:
输入:arr = [10,100,1000,10000]
输出:[10,100,10000,1000]
提示:
1 <= arr.length <= 500
0 <= arr[i] <= 10^4
这道题的思路是:根据数组中,每个元素的二进制表达中,1的个数排序,少的排在前面。如果1的个数一样就根据数字大小排序。
一、爆破法
爆破法的思路很简单,先sort,用map保存,key为二进制中1的个数,value为数组,存放数字。
最后遍历map即可(提示中给出0 <= arr[i] <= 10^4,所以只要遍历到14位即可)
执行结果如下:
77 / 77 个通过测试用例
状态:通过
执行用时: 11 ms
内存消耗: 38.8 MB
public static int[] sortByBitsMe(int[] arr) {
Map<Integer, ArrayList<Integer>> map = new HashMap<>(16, 1);
Arrays.sort(arr);
int count = 0;
for (int i : arr) {
count = binaryHaveOneNum(i);
if(map.get(count) == null) {
map.put(count,new ArrayList<>());
}
map.get(count).add(i);
}
int end = -1;
for (int i = 0; i <= 14; i++) {
ArrayList<Integer> itemList = map.get(i);
if(null == itemList) continue;
for (Integer integer : itemList) {
arr[++end] = integer;
}
}
return arr;
}
public static int binaryHaveOneNum(int num) {
int ans = 0;
for (int i = 0; i < 14; i++) {
if ((num >>> i & 1) == 1) ans++;
}
return ans;
}
这个办法很爆破,而且也很慢,但是自己毕竟经历了一天想不出来好的办法,今天只能用上愚蠢的爆破法。
下面看看评论区大佬的做法
二、位运算排序法
大佬的做法是,首先在一个高位存储数字的二进制中1的个数。
然后排序(如果1个个数相同,那么数字大小不同也能准确排序,减少了一次排序)
最后再将低位取出,重新赋值即可。
执行结果如下:
77 / 77 个通过测试用例
状态:通过
执行用时: 3 ms
内存消耗: 38.6 MB
public int[] sortByBits(int[] arr) {
for (int i = 0; i < arr.length; i++) {
arr[i] += (Integer.bitCount(arr[i]) << 16);
}
Arrays.sort(arr);
for (int i = 0; i < arr.length; i++) {
arr[i] &= 16383;
}
return arr;
}
代码量少了很多,执行时间也少了很多,内存占用差不多。果然自己对位运算还是用的不熟练啊