力扣hot100——前k个高频元素 (优先队列法,相关总结文章见另一博客)

解题思路:

哈希表记录元素出现的次数,然后将哈希表元素加入优先队列(会自动根据比较原则去排列元素)

定义优先队列根据元素频率降序排列,然后直接取出前k个元素即可

cpp 复制代码
class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
    //1.map记录元素出现的次数
        unordered_map<int,int>map;//两个int分别是元素和出现的次数
        vector<int> res;
        for(auto& c:nums){
            map[c]++;
        }


        // 采用优先队列解决
        // 此题使用自定义比较
        struct compare{
            bool operator() (pair<int,int> & p1,pair<int,int> &p2){
                return p1.second > p2.second; // 大根堆,小根堆就是小于了
            }
        };
        // priority_queue<Type, Container, Functional>;
        priority_queue<pair<int,int>,vector<pair<int,int>>,compare> pq; // 定义优先队列

        //遍历map中的元素
        //1.管他是啥,先入队列,队列会自己排序将他放在合适的位置
        //2.若队列元素个数超过k,则将栈顶元素出栈(栈顶元素一定是最小的那个)
        for(auto &m : map){
            pq.push(m);
            if(pq.size() > k){
                pq.pop();
            }
        }

        while(!pq.empty()){
            res.emplace_back(pq.top().first);
            pq.pop();
        }


        return res;

    }
};

相关推荐
Ayanami_Reii10 分钟前
进阶数据结构应用-SPOJ 3267 D-query
数据结构·算法·线段树·主席树·持久化线段树
guygg881 小时前
基于全变差的压缩感知视频图像重构算法
算法·重构·音视频
尋有緣1 小时前
力扣1327-列出指定时间段内所有的下单产品
leetcode·oracle·数据库开发
VT LI1 小时前
SDF在实时图形渲染中的核心原理与架构创新
算法·sdf·有号距离场
想七想八不如114081 小时前
408操作系统 PV专题
开发语言·算法
天一生水water1 小时前
储层认知→技术落地→产量优化
人工智能·算法·机器学习
明洞日记1 小时前
【VTK手册019】 深入理解 vtkProperty:从几何表达到 PBR 物理渲染
c++·图像处理·算法·vtk·图形渲染
Genevieve_xiao1 小时前
【数据结构与算法】【xjtuse】面向考纲学习(下)
java·数据结构·学习·算法
修炼地1 小时前
代码随想录算法训练营第二十七天 | 56. 合并区间、738.单调递增的数字、968.监控二叉树
c++·算法
仰泳的熊猫1 小时前
1031 Hello World for U
数据结构·c++·算法·pat考试