LeetCode：347.前k个高频元素

LeetCode：347.前k个高频元素
题目：
给定一个非空的整数数组，返回其中出现频率前 k 高的元素。
示例 1:

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:

输入: nums = [1], k = 1
输出: [1]

题解（c++）：
哈希+优先队列
提交结果：

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        vector<int> res;
        unordered_map<int, int> cnt;//哈希表用来计数
        priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> pq;//小根堆
        
        //将值放进哈希表并计数
        for (auto num: nums) cnt[num]++;//计数

        //把哈希表的key和value放进队列
        for (auto kv: cnt) {
            pq.push({kv.second, kv.first});//频次和值都放优先队列
            if (pq.size() > k) pq.pop(); //大于k时，队列里面最小值出列
        }
         
        //打印队列 
        while (!pq.empty()) {
            res.push_back(pq.top().second);//取出队列的值
            pq.pop();
        } 
        return res;
    }
};