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

使用优先队列实现TopK高频元素

最新推荐文章于 2025-12-04 16:02:12 发布

原创最新推荐文章于 2025-12-04 16:02:12 发布 · 225 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #数据结构 #java

该文章介绍了一种利用C++编程实现的方法，通过小顶堆（优先队列）找出给定整数数组中出现频率最高的前K个元素。首先，统计每个元素的频率，然后构建一个固定大小为K的小顶堆，依次将元素及其频率压入堆中，如果堆满则弹出最小的元素。最后，堆中的元素即为高频元素，从堆顶按序取出即可得到结果。

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