Leetcode 347. 前 K 个高频元素

Leetcode 347. 前 K 个高频元素

题目

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

思路

• 首先使用哈希表统计每一个元素出现的次数
• 定义一个小根堆，其实就是一个优先级队列，长度是k，将所有频率送入堆中，构建一个小根堆，如果长度大于k,直接出队，每次出队都是最小元素

代码

class Solution {
public:
    class mycomparison{
        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]]++;// 该元素出现的次数加一
        }
        // 对频率进行排序
        // 定义一个小根堆  大小为k
        priority_queue<pair<int,int>,vector<pair<int,int>>,mycomparison> pri_que; 

        // 用固定大小为k的小根堆
        for(unordered_map<int,int>::iterator it = map.begin();it != map.end(); it++)
        {
            pri_que.push(*it);
            if(pri_que.size() > k)
            {
                // 如果堆的大小大于k 队列弹出 保证堆的大小一直为K  维护一个优先级队列
                pri_que.pop();
            }
        }

        // 找出前k个元素，小根堆最先弹出的是最小的元素，所以倒序输出
        vector<int> result(k);
        for(int i = k - 1; i >= 0; i--)
        {
            result[i] = pri_que.top().first;// 取出队列首元素
            pri_que.pop();
        }

        return result;
    }
};