leetcode-347-前K个高频元素

//利用优先队列（注意优先队列的写法，包括创建最小堆还是最大堆，包括写不写自己的比较函数）

//pair比较时只比较第一个元素

class Solution {

public:

    vector<int> topKFrequent(vector<int>& nums, int k) {

        unordered_map<int, int> freq;  //元素，频率

        for (auto i:nums) freq[i]++;

        priority_queue<pair<int, int>, vector<pair<int,int>>, greater<pair<int, int>>> pq; //创建优先队列，维护其中只有K个元素，比较方式为less（最小堆），保持最小的先出队，频率，元素

        for (auto f:freq){

            pq.push(make_pair(f.second, f.first));

            if (pq.size() > k) pq.pop();

        }

        vector<int> res;

        while (!pq.empty()){

            res.push_back(pq.top().second);

            pq.pop();

        }

        return res;

    }

};