347. 前 K 个高频元素
class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
map<int,int> mp;
for(auto i:nums)mp[i]++;
priority_queue<pair<int,int>>q;
for(auto i:mp){
q.push({i.second,i.first});
}
vector<int>res;
while(k--){
res.push_back(q.top().second);
q.pop();
}
return res;
}
};
网友评论