前k个高频元素(优先队列)

给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。

 

示例 1:

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]

示例 2:

输入: nums = [1], k = 1
输出: [1]

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        int n = nums.size();
        map<int,int> mp;
        //建立元素与出现次数的映射关系
        for(int i=0;i<n;i++){
            mp[nums[i]]++;
        }
        //priority_queue提供了一个最大堆（默认情况下）的功能，即降序排序
        priority_queue<pair<int,int>> q;
        for(auto it = mp.begin();it!=mp.end();it++){
            q.push({it->second,it->first});
        }
        vector<int> res;
        while(k--){
            res.push_back(q.top().second);
            q.pop();
        }
        return res;
    }
};