堆--前 K 个高频元素

class Solution {
    class mycomparison {
        //小顶堆
    public:
        bool operator()(const pair<int, int>& lhs,const pair<int, int>& rhs) {
            return lhs.second > rhs.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>>,mycomparison> pri_que;

        for (auto it = map.begin(); it != map.end(); it++) {
            pri_que.push(*it);
            if (pri_que.size() > k) {
                pri_que.pop();
            }
        }

        vector<int> res(k);
        for (int i = k - 1; i >= 0; i--) {
            res[i] = pri_que.top().first;
            pri_que.pop(); 
        }
        return res;
    }
};

    public:
        bool operator()(const pair<int, int>& lhs,const pair<int, int>& rhs) {
            return lhs.second > rhs.second;
        }
    };

定义：priority_queue<Type, Container, Functional>
Type 就是数据类型，Container 就是容器类型（Container必须是用数组实现的容器，比如vector,deque等等，但不能用 list。STL里面默认用的是vector），Functional 就是比较的方式，当需要用自定义的数据类型时才需要传入这三个参数，使用基本数据类型时，只需要传入数据类型，默认是大顶堆
 

思路：首先看到求每个元素的频率，就很容易想到用map映射，难在怎么去找到数组里频率最大的元素，所以我们想到优先队列，每当求前k个最值时，可以用到它。

这串代码是构造一个小顶堆，堆的每个元素是一个map映射，当把">"改为“<”时，则构造的是一个大顶堆，用小顶堆是因为每次把小的元素pop掉后，剩下的就只剩下我们需要的前K个元素了。

注意：最后一步把优先队列赋值给vector的方式。