题目是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
(
n
l
o
g
n
)
O(nlogn)
O(nlogn),空间复杂度为
O
(
n
)
O(n)
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;
}
}
提交结果如下:
![](https://i-blog.csdnimg.cn/blog_migrate/582d0f10207f69f44ce72d2b23265a70.png)
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
提交结果如下:
![](https://i-blog.csdnimg.cn/blog_migrate/4ee54d31915c81772f466c10cd5717e1.png)