栈和队列类型题
给你一个整数数组 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 就是从小到大。
优先级队列的定义正好反过来了,可能和优先级队列的源码实现有关(我没有仔细研究),我估计是底层实现上优先队列队首指向后面,队尾指向最前面的缘故!