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] 添加鏈接描述