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

    }
};

相关推荐
格林威1 小时前
常规线扫描镜头有哪些类型?能做什么?
人工智能·深度学习·数码相机·算法·计算机视觉·视觉检测·工业镜头
程序员莫小特3 小时前
老题新解|大整数加法
数据结构·c++·算法
过往入尘土4 小时前
服务端与客户端的简单链接
人工智能·python·算法·pycharm·大模型
zycoder.4 小时前
力扣面试经典150题day1第一题(lc88),第二题(lc27)
算法·leetcode·面试
Dream it possible!4 小时前
LeetCode 面试经典 150_哈希表_存在重复元素 II(46_219_C++_简单)
leetcode·面试·散列表
蒙奇D索大4 小时前
【数据结构】考研数据结构核心考点:二叉排序树(BST)全方位详解与代码实现
数据结构·笔记·学习·考研·算法·改行学it
智驱力人工智能5 小时前
工厂抽烟检测系统 智能化安全管控新方案 加油站吸烟检测技术 吸烟行为智能监测
人工智能·算法·安全·边缘计算·抽烟检测算法·工厂抽烟检测系统·吸烟监测
学学学无无止境5 小时前
组合两个表-力扣
leetcode
程序员爱钓鱼5 小时前
Go语言实战案例——进阶与部署篇:编写Makefile自动构建Go项目
后端·算法·go
_Power_Y6 小时前
Java面试常用算法api速刷
java·算法·面试