Arrays.sort方法重写Comparator对int[]排序

首先说明,对于基本类型的数组的排序是无法重写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;
    }
}
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值