leetcode 347.前 K 个高频元素

day13

为什么使用堆？

堆就是自己排序的。。二叉树。所以使用堆更方便。同时因为题目只要求前 k 个，所以如果全部排序就 时间复杂度会更高。

大根堆还是小根堆？

小根堆。因为堆优先弹出的是 根结点。我们要求的是频率最高的。所以不能使用大根堆。不然弹出的就是最大的。

使用优先队列实现小根堆

步骤：

1. 首先用map，记录下对应的元素及其出现的次数。

2. 然后将map中的元素，依照value排序，插入优先队列中，优先队列会根据排序顺序来自己变换位置。注意：这里应该保持小根堆中的元素是 k 个，如果大于 k 就弹出！

3. 全部插入以后，由于是小根堆。所以需要逆序存放进数组。

class Solution {
public:

    class mycomparison {
    public:
        bool operator()(const pair<int, int>& lhs, const pair<int, int>& rhs) {
            return lhs.second > rhs.second;
        }
    };
    priority_queue<pair<int, int>, vector<pair<int, int>>, mycomparison > pri_que;
    vector<int> topKFrequent(vector<int>& nums, int k) {

        unordered_map<int,int> map;

        for(int i = 0; i < nums.size(); i++)
            map[nums[i]]++;

        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;
    }
};