leetcode--不同整数的最少数目

 题目是LeetCode第193场周赛的第二题，链接：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 两种整数。

 这题也挺简单的，直接贪心算法就好了，因为要使得剩下的整数种类最少，那就要尽量删去那些出现次数少的。所以需要先统计各个整数的出现次数，然后对出现次数排序，从小到大去掉k个出现次数较少的整数，剩下的整数种类就是结果。时间复杂度为$O(nlogn)$，空间复杂度为$O(n)$。

 JAVA版代码如下：

class Solution {
    public int findLeastNumOfUniqueInts(int[] arr, int k) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int a : arr) {
            map.put(a, map.getOrDefault(a, 0) + 1);
        }
        int[] values = new int[map.size()];
        int i = 0;
        for (int v : map.values()) {
            values[i++] = v;
        }
        Arrays.sort(values);
        int count = 0;
        for (i = 0; i < values.length; ++i) {
            if (values[i] <= k) {
                ++count;
                k -= values[i];
            }
        }
        return values.length - count;
    }
}

 提交结果如下:

 Python版代码如下：

class Solution:
    def findLeastNumOfUniqueInts(self, arr: List[int], k: int) -> int:
        count = collections.defaultdict(int)
        for n in arr:
            count[n] += 1
        values = list(count.values())
        values.sort()
        removeCount = 0
        for v in values:
            if v <= k:
                k -= v
                removeCount += 1
            else:
                break
        return len(values) - removeCount

 提交结果如下: