题目描述
给你一个整数数组 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;}
};