347 前k个高频元素

superior tigre2026-03-18 8:58

题目

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

示例 1：

输入：nums = $1,1,1,2,2,3$ , k = 2

输出： $1,2$

示例 2：

输入：nums = $1$ , k = 1

输出： $1$

示例 3：

输入：nums = $1,2,1,2,1,2,3,1,3,2$ , k = 2

输出： $1,2$

思路

先把数组中所有的整数放到hash map里面，key是num $i$ ，value是频率

然后桶排序，建立一个长度为n+1的数组，目的是把hash map中的key放到数组序列value下面

最后从右到左遍历找出k个key即可

代码

bash 复制代码

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> mp;
        vector<vector<int>>buck(nums.size()+1,vector<int>());
        vector <int> res;
        for(int i=0;i<nums.size();i++){
            mp[nums[i]]++;
        }
        for(auto &pair:mp){
            buck[pair.second].push_back(pair.first);
        }
        for(int i=buck.size()-1;i>0;i--){
            while(res.size()<k && !buck[i].empty()){
                res.push_back(buck[i].back());
                buck[i].pop_back();
            }
            if(res.size()==k) return res;
        }
        return res;
    }
};

注意key和value，步骤比较多，需要细心一点，不要弄反了

感谢感谢华南溜达虎力扣blind 75