排序算法桶排序 347. Top K Frequent Elements

最新推荐文章于 2025-04-21 22:09:55 发布

HuiFeiDeTuoNiaoGZ

最新推荐文章于 2025-04-21 22:09:55 发布

阅读量211

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode101

本文链接：https://blog.csdn.net/HuiFeiDeTuoNiaoGZ/article/details/118771179

LeetCode101 专栏收录该内容

101 篇文章

订阅专栏

该博客介绍了如何在给定整数数组中找出出现频率最高的k个元素。通过桶排序和哈希映射的方法，可以在O(n)的时间复杂度内解决此问题。首先，创建一个哈希表记录每个数的出现次数，然后根据出现次数构建桶，最后从高频率的桶开始收集元素，直至得到k个最频繁的元素。这种方法适用于数据分布相对均匀的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given an integer array nums and an integer k, return the k most frequent elements. You may return the answer in any order.
Example 1:
Input: nums = [1,1,1,2,2,3], k = 2
Output: [1,2]
Example 2:
Input: nums = [1], k = 1
Output: [1]
Constraints:
1 <= nums.length <= 105
k is in the range [1, the number of unique elements in the array].
It is guaranteed that the answer is unique.

桶排序是将数据划分出桶，放入有序的桶内，每个桶内分别进行排序，最后合并
桶排序的时间复杂度仅是On，非常快，但使用条件比较严格，首先数据必须能划分出桶来（比如数极差很大，接近INT_MAX，这时建有序桶就会消耗很大）
其次数据需要尽量分布均匀，不然桶排序相当于在一两个桶内排序，没意义。
讲解 https://blog.csdn.net/zihonggege/article/details/104781491

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> map;
        int max_count = 0;
        //建立每个数与出现次数的索引，并记录最多的出现次数
        for(int num : nums){
            max_count = max(max_count, ++map[num]);
        }
        //从0到max_count,每个出现次数建立一个桶
        vector<vector<int>> buckets(max_count+1);
        for(auto it : map){
            buckets[it.second].push_back(it.first);
        }
        vector<int> res;
        for(int i = buckets.size()-1; i >= 0 && res.size() < k; i--){
            for(int num : buckets[i]){
                res.push_back(num);
            }
        }
        return res;
    }
  
};