347. 前 K 个高频元素

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;
    }
};