347.前K个高频元素 C++

栈和队列类型题
给你一个整数数组 nums 和一个整数 k ,请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。

在这里插入图片描述
方法一:使用优先级队列:默认是大根堆,本题需要使用小根堆,因为需要弹出最小值,
priority_queue 优先级队列 可以操作的函数
size()
empty()
push()
pop()
top()

class Solution {
public:

class  mycomparison//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
{
    public:
        bool operator()(const pair<int,int> &left,const pair<int,int> &right)
        {
            return left.second>right.second;//小根堆
        }

};
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int>  map1;
        for(int i=0;i<nums.size();i++)
        {
            map1[nums[i]]++;
        }
        priority_queue<pair<int,int>,vector<pair<int,int>>,mycomparison>   qu;//!!!!!!!!!!!!!!!!!定义优先级队列
        unordered_map<int,int>::iterator iter;  

        for(iter=map1.begin();iter!=map1.end();iter++)
        {
            qu.push(*iter);
            if(qu.size()>k)
            {
                qu.pop();
            }
        }
        vector<int> re(k);
        for(int j=k-1;j>=0;j--)
        {
            re[j] = qu.top().first;
            qu.pop();

        }
        return re;

    }
};

方法二:

class Solution {
public:
    
    bool static cmp(const pair<int,int> &a,const pair<int,int> &b)
    {
        return a.second>b.second;
    }
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> map;
        vector<int> re;
        for(int i=0;i<nums.size();i++)
        {
            map[nums[i]]++;
        }
        vector<pair<int,int>> kk(map.begin(),map.end());
        sort(kk.begin(),kk.end(),cmp);
        for(int j=0;j<k;j++)
        {
            re.push_back(kk[j].first);
        }
        return re;
    }
};

代码随想录:

大家对这个比较运算在建堆时是如何应用的,为什么左大于右就会建立小顶堆,反而建立大顶堆比较困惑。
确实 例如我们在写快排的cmp函数的时候,return left>right 就是从大到小,return left<right 就是从小到大。
优先级队列的定义正好反过来了,可能和优先级队列的源码实现有关(我没有仔细研究),我估计是底层实现上优先队列队首指向后面,队尾指向最前面的缘故!

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值