LeetCode347：前K个高频元素

题目描述
给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。

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

//升序队列
priority_queue <int,vector<int>,greater<int> > q;
//降序队列
priority_queue <int,vector<int>,less<int> >q;

//greater和less是std实现的两个仿函数
//（就是使一个类的使用看上去像一个函数。其实现就是类中实现一个operator()，这个类就有了类似函数的行为，就是一个仿函数类了

代码

class Solution {
public:
    class cmp{
    public:
        bool operator()(const pair<int,int>& l, const pair<int,int>& r){
            return l.second > r.second;
        }
    };

    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> umap;
        for(int i = 0; i < nums.size(); ++i){
            ++umap[nums[i]];
        }

        priority_queue<pair<int, int>, vector<pair<int, int>>, cmp> pri_que;

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

        vector<int> res(k);

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