题目:
给你一个整数数组 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],新数组长度大于原数组的二分之一。
示例 2:
输入:arr = [7,7,7,7,7,7]
输出:1
解释:我们只能选择集合 {7},结果数组为空。
提示:
1 <= arr.length <= 105
arr.length
为偶数1 <= arr[i] <= 105
思路:
用哈希表记录每种元素的个数,再用优先级队列存储个数,最优优先弹出个数多的元素即可
代码:
class Solution {
public:
int minSetSize(vector<int>& arr) {
int n = arr.size();
unordered_map<int, int> hash;
priority_queue<int> pq;
for (int i = 0; i < n; i++) {
hash[arr[i]]++;
}
for (auto it : hash) {
pq.push(it.second);
}
int target = n % 2 == 1 ? n / 2 + 1 : n / 2;
int cur = 0, ans = 0;
while (!pq.empty()) {
int top = pq.top();
pq.pop();
cur += top;
ans++;
if (cur >= target) return ans;
}
return -1;
}
};