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;
    }
};