LeetCode#347. 前 K 个高频元素

给定一个非空的整数数组，返回其中出现频率前 k 高的元素。

示例 1:

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

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

提示：

你可以假设给定的 k 总是合理的，且 1 ≤ k ≤ 数组中不相同的元素的个数。
你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。
题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的。
你可以按任意顺序返回答案。
通过次数57,740提交次数95,240

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/top-k-frequent-elements
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解法：通过map来统计次数，通过大小为k的小顶堆来保存最高频的k个数，通过与堆顶比较来更新，最后从堆中输出即可

#include <iostream>
using namespace std;
class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        //思路：构建一个大小为k的小顶堆，用来保存k个最高频的词，每次判断新的词的时候
        //就通过与堆顶比较，更高频就换，否则就不加入
        vector<int> ret;
        //需要先统计一下词频,用map来统计
        map<int,int> m;
        using E = std::pair<int, int>;
        std::priority_queue<E, std::vector<E>, std::greater<E>> minHeap;
        for(auto &n : nums){
            if(m.find(n) != m.end()){
                //说明已经存在
                m[n]++;
            }else{
                //说明不存在
                m[n] = 1;
            }
        }
        for(auto &item : m){
            auto pair = make_pair(item.second,item.first);
            if(k > 0){
                //说明可以直接加入堆
                minHeap.push(pair);
                k--;
            }else{
                //如果此时已经满了,则需要根据堆顶来判断
                if(pair.first > minHeap.top().first){
                    //如果频率比堆顶高，则换一下
                    minHeap.pop();
                    minHeap.push(pair);
                }
                //否则就不用管啦
            }
        }
        //从堆中导出
        while(!minHeap.empty()){
            ret.push_back(minHeap.top().second);
            minHeap.pop();
        }
        return ret;
    }
};