数组大小减半

2024-12-15

题目:

给你一个整数数组 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;
    }
};