给你一个整数数组 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
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/least-number-of-unique-integers-after-k-removals 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution { //C++
public:
int findLeastNumOfUniqueInts(vector<int>& arr, int k) {
unordered_map<int,int> m;
int i;
for(i = 0; i < arr.size(); ++i)
m[arr[i]]++;
vector<int> count(m.size());
i = 0;
for(auto it = m.begin(); it != m.end(); ++it)
count[i++] = it->second;
sort(count.begin(), count.end());
i = 0;
while(k > 0)
{
k -= count[i];
if(k >= 0)
i++;
}
return count.size()-i;
}
};
432 ms 61.2 MB
class Solution:# py3
def findLeastNumOfUniqueInts(self, arr: List[int], k: int) -> int:
d = {}
for i in arr:
if i in d:
d[i] += 1
else:
d[i] = 1
count = list(d.values())
# count.sort()
count = sorted(count)
i = 0
for c in count:
k -= c
if k >= 0:
i += 1
if k <= 0:
break
return len(count) - i
120 ms 26.8 MB