代码随想录算法训练营第13天|LeetCode347.前k个高频元素
LeetCode347.前k个高频元素
题目:347. 前 K 个高频元素 - 力扣(LeetCode)
文章:代码随想录 (programmercarl.com)
视频:优先级队列正式登场!大顶堆、小顶堆该怎么用?| LeetCode:347.前 K 个高频元素_哔哩哔哩_bilibili
思路
- 频率:使用map存放元素出现的频率。(key,value)其中key是元素,value是出现的次数。
- 排序:
使用大根堆/小根堆得到前k个结果。所有map都遍历进堆,但是堆里面只维护k个元素。
想当然地以为使用大根堆,但实际上push进一个元素,需要pop一个元素,而从根开始pop的话就会把最大值pop出去,最后堆里面剩下的实际上是前k个低频元素。因此需要使用小根堆。时间复杂度O(nlogk) - 实现:优先级队列,就是堆的结构。
- 若要按从大到小输出,那么倒序遍历result数组即可。
代码
问题:重载运算符为什么要那样写?放入小根堆的代码为什么我写的有问题?
class Solution {
public:
//注意这里自己写的比较函数,不知道为什么要这种写法?不可以直接定义函数吗?
class mycomparison {
public:
bool operator()(const pair<int, int>& lhs, const pair<int, int>& rhs) {
return lhs.second > rhs.second;
}
};
vector<int> topKFrequent(vector<int>& nums, int k) {
//计算频率
unordered_map<int,int> map;
for(int i=0;i<nums.size();i++){
map[nums[i]]++;
}
//放入小根堆
priority_queue< pair<int,int>, vector<pair<int,int> >, mycomparison > que;
//下面注释的方法写的是错的,为什么呢?
// for(int i=0; i<nums.size(); i++){
// que.push(make_pair(nums[i],map[nums[i]]));
// if(que.size()>k)
// que.pop();
// }
for (unordered_map<int, int>::iterator it = map.begin(); it != map.end(); it++) {
que.push(*it);
if (que.size() > k) { // 如果堆的大小大于了K,则队列弹出,保证堆的大小一直为k
que.pop();
}
}
//从高到低输出
vector<int> res(k);
//注意这里是k-1
for(int i=k-1 ; i>=0 ;i--){
res[i]=que.top().first;
que.pop();
}
return res;
}
};
总结
- 大根堆、小根堆——使用情景:前k个
- pair的定义与使用:pair<int,int>,p.first,p.second
- 优先级队列的定义、使用