首先说明,对于基本类型的数组的排序是无法重写Comparator来进行排序的,这是因为基本类型不是Object,更不会实现Comparator接口,自然也就无法调用compareTo方法来进行比较大小。因此若想使用简单的方法自定义排序方式,只能将基本类型转化为对应的包装类来进行自定义排序。
多嘴一句,如果想要使代码更简洁优雅,可以使用lambda表达式来代替匿名内部类。
LeetCode第1342题
给你一个整数数组 arr。你可以从中选出一个整数集合,并删除这些整数在数组中的每次出现。
返回 至少 能删除数组中的一半整数的整数集合的最小大小。
示例 1:
输入:arr = [3,3,3,3,5,5,5,2,2,7]
输出:2
解释:选择 {3,7} 使得结果数组为 [5,5,5,2,2]、长度为 5(原数组长度的一半)。
大小为 2 的可行集合有 {3,5},{3,2},{5,2}。
选择 {2,7} 是不可行的,它的结果数组为 [3,3,3,3,5,5,5],新数组长度大于原数组的二分之一。
考虑用一个map记录每个数字出现的次数,再将数字按出现次数排序,一种思路是构建一个 M a p . E n t r y < k e y , v a l u e > Map.Entry<key, value> Map.Entry<key,value>的数组,另一种是构建key的int型数组,显然第二种方式更省内存,因此选择第二种,而在我重写Arrays.sort方法时
int[] arr = new int[len];
Arrays.sort(arr, new Comparator<Integer>(){
public int compare(Integer a, Integer b) {
return new Integer(map.get(a)).compareTo(new Integer(map.get(b)));
}
});
发现报错,看了报错后发现应该是自动装箱/拆箱出了问题,也就是说,Arrays.sort重写Comparator,不可以用基本类型数组,当然也就不能用int[]数组,要想对int型排序的话,必须用Integer[]数组才可以。另一种方法是用List排序,如下
List<Integer> list = new ArrayList();
for(int i : map.keySet())
list.add(i);
Collections.sort(list, new Comparator<Integer>(){
public int compare(Integer a, Integer b) {
return new Integer(map.get(a)).compareTo(new Integer(map.get(b)));
}
});
这样就没有再报错。
完整代码如下
class Solution {
public int minSetSize(int[] arr) {
HashMap<Integer, Integer> map = new HashMap();
for(int i : arr) {
if(map.containsKey(i)) {
map.put(i, map.get(i) + 1);
} else {
map.put(i, 1);
}
}
List<Integer> list = new ArrayList();
for(int i : map.keySet())
list.add(i);
Collections.sort(list, new Comparator<Integer>(){
public int compare(Integer a, Integer b) {
return new Integer(map.get(a)).compareTo(new Integer(map.get(b)));
}
});
int sum = 0;
//System.out.println(list);
for(int i = list.size() - 1; i >= 0; i--) {
sum += map.get(list.get(i));
if(sum >= arr.length / 2)
return list.size() - i;
}
return 0;
}
}