代码随想录训练营 栈与队列

前K个高频元素

优先队列（小顶堆）

class Solution {
public:

    // 构建小顶堆的比较规则
    //这里对（）进行了重载，返回的是较大
    class mycomparison {
        public:
            bool operator()(const pair<int,int> & lhs, const pair<int,int> & rhs) {
                return lhs.second > rhs.second;
            }
    };

    
    vector<int> topKFrequent(vector<int>& nums, int k) {
           //统计元素出现的频率
        unordered_map<int,int> map;
        for(int i = 0; i < nums.size(); i++){
            map[nums[i]]++;
        }
    
    
    // 定义一个小顶堆，大小为k
    priority_queue<pair<int,int>, vector<pair<int, int>>, mycomparison> pri_que;

   // 用固定大小为k的小顶堆，扫面所有频率的数值
        for (unordered_map<int, int>::iterator it = map.begin(); it != map.end(); it++) {
            pri_que.push(*it);
            if (pri_que.size() > k) { // 如果堆的大小大于了K，则队列弹出，保证堆的大小一直为k
                pri_que.pop();
            }
        }

    vector<int> result(k);
    for (int i =k - 1; i >= 0; i--){
        result[i] = pri_que.top().first;
        pri_que.pop();
    }
    return result;
    }
};