5.24大顶堆，小顶堆和重载（347. 前 K 个高频元素）

347. 前 K 个高频元素

class Solution {
    //此题设计一个优先队列来做，优先队列会把入队的元素进行排序，大的排在前面，小的排在后面
    //并且我们可以重写优先队列的比较方式
    struct comp{
        //一定注意书写的时候，跟平时的比较顺序是反的
        bool operator()(const pair<int, int> left, const pair<int, int> right){
            return left.second < right.second;
        }
    };
public:
    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>>, comp> pri_que;
       for(auto m:map){
           pri_que.push(m);
       }

        vector<int> result;
       for(int i = 0; i < k; i++){
           result.push_back(pri_que.top().first);
           pri_que.pop();
       }

       return result;


    }
};