一道题带你熟练优先队列:前k个高频元素

1.347. 前 K 个高频元素 - 力扣(LeetCode)

2.题目中前k个,就很明显的提示了我们这是堆的topk问题,所以要使用优先队列,而这种统计频率的题肯定要使用哈希表去重,所以用了unordered_map,由于优先队列默认是大堆,而我们要建大堆,但是元素不是简单的数据,而是pair的second进行比较,所以还是要采取(数据类型,存储容器,自定义比较类/比较函数)的格式

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
unordered_map<int,int>map1;
for(auto a:nums)
{
    map1[a]++;
}
struct t
{
bool operator()(pair<int,int>&p1,pair<int,int>&p2)
{
    return p1.second>p2.second;
}
};
priority_queue<pair<int,int>,vector<pair<int,int>>,t>q;
for(auto& b:map1)
{
    q.push(b);
    if(q.size()>k)
    {
        q.pop();
    }
}
vector<int>ret;
while(!q.empty())
{
    ret.push_back(q.top().first);
    q.pop();
}
return ret;
    }
};

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值