1338. 数组大小减半 ●●

1338. 数组大小减半 ●●

描述

给你一个整数数组 arr。你可以从中选出一个整数集合，并在 arr 中删除这些整数。

返回 至少 能删除数组中的一半整数的整数集合的最小大小。

示例

输入：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]，新数组长度大于原数组的二分之一。

题解

1. 哈希计数 + 贪心

1. 哈希表统计各元素出现的次数；
2. 创建一个下标为出现次数的数组，记录出现了该次数的元素个数；
3. 从次数最大往最小遍历，每次删除出现次数最大的元素，进行判断。

• 时间复杂度：O(N)，其中 N 是数组 arr 的长度。
• 空间复杂度：O(N)。

class Solution {
public:
    int minSetSize(vector<int>& arr) {
        unordered_map<int, int> map;
        int maxCnt = 0, n = arr.size();
        for(int num : arr){
            ++map[num]; 
            maxCnt = max(maxCnt, map[num]);     // 统计出现次数
        }
        if(maxCnt >= (n-1)/ 2 + 1) return 1;    // 最大次数满足，直接返回 1
        vector<int> cnts(maxCnt+1, 0);          // 桶，下标即为频次，元素值为该频次的元素个数
        for(auto iter = map.begin(); iter != map.end(); ++iter){
            ++cnts[iter->second];               
        }
        int cnt = 0, ans = 0;
        for(int i = maxCnt; i >= 0; --i){       // 从最大的桶往前遍历
            while(cnts[i] > 0){
                cnt += i;                       // 删除当前最大的次数
                --cnts[i];
                ++ans;
                if(cnt >= (n-1)/ 2 + 1) return ans;
            }
        }
        return ans;
    }
};