2024-03-14 18:24:34
目录
给定一个非空的整数数组,返回其中出现频率前 k 高的元素。
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
输入: nums = [1], k = 1
输出: [1]
说明:
#define PR pair<int, int> // number and its frequency
class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
unordered_map<int, int> mp;
for(int num : nums){
if(mp.find(num) == mp.end()){
mp[num] = 1;
}else{
mp[num]++;
}
}
vector<PR> lst;
for(auto it : mp){
lst.push_back({-it.second, it.first});
}
sort(lst.begin(), lst.end());
vector<int> res;
for(int i = 0; i < k; i++){
res.push_back(lst[i].second);
}
return res;
}
};