有时候用arrays.sort或者collections.sort的时候需要重写比较器来按照我们想要的比较方式对某些元素进行比较或者排序,例如leetcode题目1356中就需要重写比较器。
题目描述
将数组中的元素按照其二进制表示中数字 1 的数目升序排序。
如果存在多个数字二进制中 1 的数目相同,则必须将它们按照数值大小升序排列。
重写构造器代码
Collections.sort(list, new Comparator<Integer>(){
//重写比较函数
@Override
public int compare(Integer o1, Integer o2) {
if (bit[o1] != bit[o2]){
return bit[o1] - bit[o2];
}
else
return o1 - o2;
}
});
其中bit[i]存放的是数字对应的二进制中1的数量,代码中当o1和o2数字不相同时,比较他们对应二进制中1的数量,相同时,比较其值的大小。
题目中需要先把数组转为list<Integer>,因为int类型无法进行重构,所以先转成Integer。
t同样,也可以用一个Integer的数组代替前面说的列表
import java.util.*;
public class Solution {
public static int get(int x){
int res = 0;
while (x != 0){
res += x %2;
x /= 2;
}
return res;
}
public static int[] sortByBits(int[] arr) {
int[] bit = new int[10001];
for (int x : arr){
bit[x] = get(x);
}
Integer[] arrs = new Integer[arr.length];
for (int i = 0; i < arr.length; i++){
arrs[i] = arr[i];
}
Arrays.sort(arrs, new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
if (bit[o1] != bit[o2]){
return bit[o1] - bit[o2];
}
else
return o1 - o2;
}
});
for (int i = 0; i < arr.length; i++){
arr[i] = arrs[i];
}
return arr;
}
public static void main(String[] args){
int[] newInterval = {1024,512,256,128,64,32,16,8,4,2,1};
int[] ans = sortByBits(newInterval);
// System.out.println(ans.length);
for (int i = 0; i < ans.length; i++) {
System.out.println(ans[i] + " ");
}
}
}