您的位置:首页 > 房产 > 家装 > LeetCode347:前K个高频元素

LeetCode347:前K个高频元素

2025/3/4 6:08:06 来源:https://blog.csdn.net/m0_50127633/article/details/139736556  浏览:    关键词:LeetCode347:前K个高频元素

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

在这里插入图片描述
解题思想
使用优先队列
priority_queue<Type, Container, Functional>
Type 就是数据类型,Container 就是容器类型(Container必须是用数组实现的容器,比如vector,deque等等,但不能用 list。STL里面默认用的是vector),Functional 就是比较的方式,当需要用自定义的数据类型时才需要传入这三个参数,使用基本数据类型时,只需要传入数据类型,默认是大顶堆

//升序队列
priority_queue <int,vector<int>,greater<int> > q;
//降序队列
priority_queue <int,vector<int>,less<int> >q;//greater和less是std实现的两个仿函数
//(就是使一个类的使用看上去像一个函数。其实现就是类中实现一个operator(),这个类就有了类似函数的行为,就是一个仿函数类了

代码

class Solution {
public:class cmp{public:bool operator()(const pair<int,int>& l, const pair<int,int>& r){return l.second > r.second;}};vector<int> topKFrequent(vector<int>& nums, int k) {unordered_map<int,int> umap;for(int i = 0; i < nums.size(); ++i){++umap[nums[i]];}priority_queue<pair<int, int>, vector<pair<int, int>>, cmp> pri_que;for(auto it = umap.begin(); it != umap.end(); ++it){pri_que.push(*it);if(pri_que.size() > k) pri_que.pop();}vector<int> res(k);for(int i = 0; i < k; i++){res[i] = pri_que.top().first;pri_que.pop();}return res;}
};

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com