leetcode 5437. 不同整数的最少数目（C++）

---

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

C++

class Solution {
public:
    int findLeastNumOfUniqueInts(vector<int>& arr, int k) 
    {
        int n=arr.size();
        map<int,int> tmp;
        for(int i=0;i<n;i++)
        {
            tmp[arr[i]]++;
        }
        vector<int> vec;
        for(auto it:tmp)
        {
            vec.push_back(it.second);
        }
        sort(vec.begin(),vec.end());
        int idx=0;
        for(int i=0;i<vec.size();i++)
        {
            if(vec[i]<k)
            {
                k-=vec[i];
            }
            else if(vec[i]==k)
            {
                idx=i+1;
                break;
            }
            else
            {
                idx=i;
                break;
            }
        }
        return vec.size()-idx;
    }
};