力扣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;

    }
};

相关推荐
2501_901147833 分钟前
四数相加问题的算法优化与工程实现笔记
笔记·算法·面试·职场和发展·哈希算法
亿秒签到6 分钟前
第六届传智杯程序设计国赛B组T4·小苯的字符串染色
数据结构·算法·传智杯
chao1898446 分钟前
基于字典缩放的属性散射中心参数提取算法与MATLAB实现
开发语言·算法·matlab
石去皿12 分钟前
大厂AI算法面试题汇总
人工智能·算法
晚风吹长发24 分钟前
初步了解Linux中的信号保存和简单使用
linux·运维·服务器·数据结构·后端·算法
AndrewHZ29 分钟前
【图像处理与ISP技术】图像格式与存储原理
图像处理·算法·isp·图像压缩·图像格式·图像存储·图像信号处理
仰泳的熊猫30 分钟前
题目1431:蓝桥杯2014年第五届真题-分糖果
数据结构·c++·算法·蓝桥杯
Fᴏʀ ʏ꯭ᴏ꯭ᴜ꯭.31 分钟前
Haproxy负载均衡算法全解析
算法
永远都不秃头的程序员(互关)3 小时前
【决策树深度探索(四)】揭秘“混乱”:香农熵与信息纯度的量化之旅
算法·决策树·机器学习
永远都不秃头的程序员(互关)3 小时前
【决策树深度探索(三)】树的骨架:节点、分支与叶子,构建你的第一个分类器!
算法·决策树·机器学习