《每日一题》347. 前 K 个高频元素

class Solution {
public:
    //小顶堆，左大于右
    class comparison{
    public:
        bool operator()(const pair<int, int>& lhs, const pair<int, int>& rhs){
            return lhs.second > rhs.second;
        }
    };
    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>>, comparison> pri_que;//定义<>只需要类型,第二个是map的数据类型
        vector<int> result(k);//设定vector大小,否侧在for循环给i的值时不能确定大小，因为是反过来的
        for (unordered_map<int, int>::iterator it = map.begin(); it != map.end(); it++){//for循环的写法
            pri_que.push(*it);
            if (pri_que.size() > k){//超过k就要开始舍弃
                pri_que.pop();
            }
        }
        //队列里只有k个数，最小的在堆口，因此要反过来
        for (int i = k - 1; i >= 0; i--){
            result[i] = pri_que.top().first;//弹出的数值，不是频率次数
            pri_que.pop();
        }
        return result;
    }
};