题目:
给你一个整数数组 nums
和一个整数 k
,请你返回其中出现频率前 k
高的元素。你可以按 任意顺序 返回答案。
方法:
分析题目意图,我们可以将其拆解为三个部分。
1:首先先统计数组中每个元素出现的频率,这步我们可以使用hash的map来完成,前面题目用过
2:对每个元素出现的频率进行排序。
3:输出前K个出现频率最高的元素。
后两步我们可以使用优先级队列来完成,所谓优先级队列是一个队列,他只提供了两个方法:从队首出队和从队尾进队,但优先级队列内部元素是自动依照元素的权值排列的。其底层实现方式其实是堆的结构,我们可以将获得数组元素出现频次的map进行入队操作。其可以自动对输入的元素频次进行排序,当队列元素超过K时,我们就将频次低的元素出队,因此我们可以使用小顶堆的实现方式,每次出来的就是频次最低的元素。最后全部遍历完成之后,队列内剩下的就是出现频率最高的k个元素了,我们按照逆序顺序(因为是小顶堆,最高频次的在后面)输出即可。
题解:
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> pri_que;
for (unordered_map<int, int>::iterator it = map.begin(); it != map.end(); it++) {
pri_que.push(*it);
if (pri_que.size() > k) { // 如果堆的大小大于K,则出队,保证堆的大小一直为k
pri_que.pop();
}
}
// 找出前K个高频元素,因为小顶堆先弹出的是最小的,所以倒序来输出到数组
vector<int> result(k);
for (int i = k - 1; i >= 0; i--) {
result[i] = pri_que.top().first;
pri_que.pop();
}
return result;
}
};