Leetcode 347. 前K个高频元素 解题思路及C++实现

方法一：最小堆

解题思路：

先使用一个unordered_map来遍历nums容器，得到每个元素对应的频数。

再使用最小堆，对unordered_map中的频数进行遍历，得到k个最大的频数对应的<频数，元素>对。

最后将最小堆中的元素push_back进结果容器res中。

class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
unordered_map<int, int> mp;  //unordered_map中存储的是<元素，频数>
for(auto num: nums) mp[num]++;

priority_queue< pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>> > pq; //最小堆，存储的是<频数， 元素>
for(auto val: mp){
pq.push(make_pair(val.second, val.first));
if(pq.size() > k) pq.pop();
}
//将最小堆中的元素放在res容器中，并返回结果
vector<int> res;
for(int i = 0; i < k; i++){
res.push_back(pq.top().second);
pq.pop();
}
return res;
}
};