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

    }
};

相关推荐
鱼很腾apoc4 小时前
【学习篇】第20期 超详解 C++ 多态:从语法规则到底层原理
java·c语言·开发语言·c++·学习·算法·青少年编程
小许同学记录成长6 小时前
三维重建技术文档
算法·无人机
小O的算法实验室7 小时前
2026年ASOC,基于多目标优化去噪双存档进化算法+路径规划,深度解析+性能实测
算法·论文复现·智能算法·智能算法改进
2601_954526757 小时前
逆向解析Temu底层动销算法:基于API高并发轮询与全域存量透视的自动化架构重构
算法·架构·自动化
Σίσυφος19008 小时前
数据标准化(拟合的时候使用非常重要)
人工智能·算法
knight_9___8 小时前
大模型project面试7
人工智能·python·算法·面试·大模型·agent
NashSKY9 小时前
EM 算法完整推导与本质剖析
算法·机器学习·概率论
foundbug9999 小时前
MATLAB实现:基于图像对比度和波段相关性的高光谱波段选择算法
开发语言·算法·matlab
嘿嘿嘿x39 小时前
Linux-实践
linux·运维·算法