LeetCode : Top K Frequent Elements

最新推荐文章于 2024-10-20 08:13:14 发布

IMapMap

最新推荐文章于 2024-10-20 08:13:14 发布

阅读量296

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/IMapMap/article/details/53434780

LeetCode 专栏收录该内容

25 篇文章

订阅专栏

本文介绍了一种基于哈希表和桶排序思想的高效算法，用于找出数组中出现频率最高的k个元素。通过统计每个数字出现的次数并将其存放在对应频率的桶中，最后从高频率桶开始收集结果。

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

Solution - O(n)

vector<int> topKFrequent(vector<int>& nums, int k)
{
    // 统计每个数字的频率
    unordered_map<int, int> rec;
    for(int i = 0; i < nums.size(); i++)
        rec[nums[i]]++;

    // 按频率存放数字
    vector<vector<int>> bucket(nums.size());
    for (auto it : rec)
        bucket[it.second - 1].push_back(it.first);

    vector<int> res;
    for (int i = bucket.size() - 1; i >= 0 && k > 0; i--)
    {
        for (int temp : bucket[i])
            res.push_back(temp);
        k -= bucket[i].size();
    }

    return res;
}