代码随想录算法训练营第14天|LeetCode347.前k个高频元素

本文链接：https://blog.csdn.net/weixin_63948917/article/details/136491237

代码随想录算法训练营第13天|LeetCode347.前k个高频元素

LeetCode347.前k个高频元素

题目：347. 前 K 个高频元素 - 力扣（LeetCode）
文章：代码随想录 (programmercarl.com)
视频：优先级队列正式登场！大顶堆、小顶堆该怎么用？| LeetCode：347.前 K 个高频元素_哔哩哔哩_bilibili

在这里插入图片描述

思路

频率：使用map存放元素出现的频率。（key，value）其中key是元素，value是出现的次数。
排序：
使用大根堆/小根堆得到前k个结果。所有map都遍历进堆，但是堆里面只维护k个元素。
想当然地以为使用大根堆，但实际上push进一个元素，需要pop一个元素，而从根开始pop的话就会把最大值pop出去，最后堆里面剩下的实际上是前k个低频元素。因此需要使用小根堆。时间复杂度O(nlogk)
实现：优先级队列，就是堆的结构。
若要按从大到小输出，那么倒序遍历result数组即可。

代码

问题：重载运算符为什么要那样写？放入小根堆的代码为什么我写的有问题？

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]]++;
        }
        //放入小根堆
        priority_queue< pair<int,int>, vector<pair<int,int> >, mycomparison > que;
        //下面注释的方法写的是错的，为什么呢？
        // for(int i=0; i<nums.size(); i++){
        //     que.push(make_pair(nums[i],map[nums[i]]));
        //     if(que.size()>k)
        //         que.pop();
        // }
        for (unordered_map<int, int>::iterator it = map.begin(); it != map.end(); it++) {
            que.push(*it);
            if (que.size() > k) { // 如果堆的大小大于了K，则队列弹出，保证堆的大小一直为k
                que.pop();
            }
        }
        //从高到低输出
        vector<int> res(k);
        //注意这里是k-1
        for(int i=k-1 ; i>=0 ;i--){
            res[i]=que.top().first;
            que.pop();
        }
        return res;

    }
};