刷题日记（Day12）

优先级队列（小顶堆）经典问题 347. 前 K 个高频元素

leetcode-347. 前 K 个高频元素

题目描述：

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

示例 1:

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]

示例 2:

输入: nums = [1], k = 1
输出: [1]

遇到求前多少个高频元素的问题，首先是肯定维护一个哈希表来记录当前所有元素的出现频率，然后对这个哈希表在出现频率上进行排序，然而即使使用快速排序，时间复杂度也是O(n*logn)，此时最优的方法是使用堆结构。

堆结构其实就是一颗完全二叉树（除了最后一层之外都被填满的二叉树），堆的实现一般可以采用线性表，或者直接用二叉树。堆的特性是这样的：大顶堆（大根堆）要求父节点的值要大于左右孩子节点，小顶堆（小根堆）则是要求父节点的值小于左右的孩子节点。

结合题意，我们要保留最大的K个节点，每次弹出时就要弹出最小的节点。因此应该选用小顶堆来实现。
代码如下，使用STL中的优先级队列实现小顶堆(注意比较函数的实现，与sort是相反的)：

// 小顶堆
class my_compare {
public:
    bool operator()(const pair<int, int>& lhs, const pair<int, int>& rhs) {
        return lhs.second > rhs.second;
    }
};

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int, int> map;
        for(int & n : nums){
            map[n]++;
        }

        //定义一个小顶堆（用优先级队列来实现）
        //需要实现自定义的比较函数，
        priority_queue<pair<int, int>, vector<pair<int, int>>, my_compare> pri_que;

        for(auto &m : map){
            pri_que.push(m);
            if(pri_que.size() > k){
                pri_que.pop();
            }
        }

        vector<int> result;
        while(!pri_que.empty()){
            result.push_back(pri_que.top().first);
            pri_que.pop();
        }
        return result;
    }
};