Leetcode: 1481. 不同整数的最少数目

题目

给你一个整数数组 arr 和一个整数 k 。现需要从数组中恰好移除 k 个元素，请找出移除后数组中不同整数的最少数目。

示例 1：

输入：arr = [5,5,4], k = 1
输出：1
解释：移除 1 个 4 ，数组中只剩下 5 一种整数。
示例 2：

输入：arr = [4,3,1,1,3,3,2], k = 3
输出：2
解释：先移除 4、2 ，然后再移除两个 1 中的任意 1 个或者三个 3 中的任意 1 个，最后剩下 1 和 3 两种整数。
 

提示：

1 <= arr.length <= 10^5
1 <= arr[i] <= 10^9
0 <= k <= arr.length

分析

完成这道题目需要两个步骤，首先要统计每个数字出现的次数，其次要完成按出现次数的排序，因为先移除出现次数少的数字会移除更多数字。所以自然想到利用 TreeMap。但是，TreeMap 在设计时只允许对 key 进行排序，而我们需要对 value 进行排序，所以这道题的难点在于如何按照 value 值进行排序。

import java.util.Map.Entry;
class Solution {
    public int findLeastNumOfUniqueInts(int[] arr, int k) {
        // item, 次数
        Map<Integer, Integer> map = new HashMap<>();
        for (int a : arr) {
            int tmp = map.getOrDefault(a, 0);
            map.put(a, tmp + 1);
        }
        
        List<Entry<Integer, Integer>> list = new ArrayList<>(map.entrySet());
        
        Collections.sort(list, (o1, o2) -> (o1.getValue().compareTo(o2.getValue())));
        
        for (Entry<Integer, Integer> e : list) {
            if (e.getValue() <= k) {
                k -= e.getValue();
                map.remove(e.getKey());
            } else {
                break;
            }
        }
        return map.keySet().size();
    }
}

其中，Map.Entry 是 Map 声明的一个内部接口，定义为 Entry<K, V>, 其中有 getKey() 和 getValue() 方法。在这种情况下，可以将键值对放入 List 中，然后利用 Collections.sort 方法按 value 进行排序。当然在该种情况下，使用 HashMap 就可以了，因为并不需要对 key 进行排序。

另外，注意到，题目不需要知道哪些数字留下了，哪些数字被删除了。所以可以将数字出现的次数放入数组中，利用 Array.sort 进行排序，然后按出现次数从小到大进行删除。

class Solution {
    public int findLeastNumOfUniqueInts(int[] arr, int k) {
        // item, 次数
        Map<Integer, Integer> map = new HashMap<>();
        for (int a : arr) {
            int tmp = map.getOrDefault(a, 0);
            map.put(a, tmp + 1);
        }
        
        int n = map.keySet().size();
        int[] cnt = new int[n];
        int index = 0;
        for (int key : map.keySet()) {
            cnt[index++] = map.get(key);
        }

        Arrays.sort(cnt);

        index = 0;
        while (n > 0 && k >= cnt[index]) { // n == 0 说明当前已经删完了
            k -= cnt[index++];
            n--;
        }
        return n;
    }
}