Leetcode 1481. 不同整数的最少数目

问题描述

给你一个整数数组 arr 和一个整数 k 。现需要从数组中恰好移除 k 个元素，请找出移除后数组中不同整数的最少数目。

示例 1：

输入：arr = [5,5,4], k = 1
输出：1
解释：移除 1 个 4 ，数组中只剩下 5 一种整数。

解题报告

问题的关键是对 unordered_map 按照 value 进行排序。

方法是将 unordered_map 的 key 和 value 作为 pair 放到 vector 中，然后定义 pair 的排序方法。

static bool cmp(pair<int, int>a, pair<int, int>b){
    return a.second<b.second;//按照第二个元素从小到大排序
}

实现代码

class Solution {
public:
    static bool cmp(pair<int, int>a, pair<int, int>b){
        return a.second<b.second;
    }
    
    int findLeastNumOfUniqueInts(vector<int>& arr, int k) {
        unordered_map<int, int>mp;
        for(auto num:arr) mp[num]++;
        
        vector<pair<int, int>>v;
        for(auto i=mp.begin();i!=mp.end();i++) 
        	v.push_back(pair<int, int>(i->first, i->second));
        sort(v.begin(), v.end(), cmp);
        
        int counter=0,i;
        for(i=0;i<v.size();i++){
            int tmp=k-counter;
            if(tmp>=v[i].second)counter+=v[i].second;
            else break;
        }
        return v.size()-i;
    }
};

参考资料

[1] 添加链接描述