想了解更多数据结构以及算法题,可以关注微信公众号“数据结构和算法”,每天一题为你精彩解答。也可以扫描下面的二维码关注
问题描述
给你一个整数数组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的个数从大到小排序,如果1的个数相同,就按照原来数字的大小进行排序。
我们仔细看这题有个限制条件就是0<=arr[i]<=10^4,也就是说数组元素都是非负数,并且都是小于等于10000的。不知道大家有没有学Android的,看过Android源码,你会看到里面经常会用一个数字表示多种状态,有的是用二进制中某一位来表示,有的是用二进制中的多位来表示,看到这里是不是也有了灵感。
所以一种最简单的方式,先计算出数组中每个元素二进制中1的个数乘以100000+原来的数,让他成为一个新的数,接着再对他们进行排序,最后再还原,看下代码
public int[] sortByBits(int[] arr) {
int length = arr.length;
//数组中每个数字的二进制位乘以100000再加上原来的数值,
//成为一个新的数
for (int i = 0; i < length; i++) {
arr[i] = Integer.bitCount(arr[i]) * 100000 + arr[i];
}
//对这个新的数进行排序
Arrays.sort(arr);
//然后再把新的数字还原成原来的数字
for (int i = 0; i < length; i++) {
arr[i] %= 100000;
}
return arr;
}
先计算再排序
还有一种方式就是先计算数组中每个数字二进制中1的个数,然后再排序,这种就比较简单了。关于一个数字二进制中1的个数可以看下425,剑指 Offer-二进制中1的个数,这里列出了18种方式,具体细节可以看下
关于排序,在两年前写过十几种排序算法,也可以看下
如果这样两者结合的话,那么答案就比较多了,我们来随便挑一组组合。排序就使用插入排序,计算二进制中1的个数我们随便挑一个,答案如下
public int[] sortByBits(int[] arr) {
//二维数组,temp[i][0]存储的是当前的值
//temp[i][1]存储的是当前值的二进制中1的个数
int[][] temp = new int[arr.length][2];
for (int i = 0; i < arr.length; i++) {
temp[i][0] = arr[i];
temp[i][1] = hammingWeight(arr[i]);
}
//使用插入排序
insertSort(temp);
for (int i = 0; i < arr.length; i++) {
arr[i] = temp[i][0];
}
return arr;
}
//插入排序
private void insertSort(int[][] array) {
for (int i = 1; i < array.length; i++) {
int j;
int[] temp = array[i];
for (j = i - 1; j >= 0; j--) {
//先比较位1的大小,如果相同再比较数字的大小
if (array[j][1] > temp[1] || (array[j][1] == temp[1] && array[j][0] > temp[0])) {
array[j + 1] = array[j];//往后挪
} else {
break;//没有交换就break
}
}
array[j + 1] = temp;
}
}
//计算二进制中1的个数
public int hammingWeight(int n) {
int count = 0;
while (n != 0) {
n &= n - 1;
count++;
}
return count;
}
问题分析
这题结合了计算二进制中1的个数和排序两种方式,这两种方式之前都有大量的介绍,如果结合写的话,答案还是比较多的。