347. Top K Frequent Elements

最新推荐文章于 2024-06-11 11:37:12 发布

ssevenstar

最新推荐文章于 2024-06-11 11:37:12 发布

阅读量265

点赞数

CC 4.0 BY-SA版权

分类专栏： algorithm leetcode 文章标签： topK bucketSort

本文链接：https://blog.youkuaiyun.com/ssevenstar/article/details/72783280

algorithm 同时被 2 个专栏收录

26 篇文章

订阅专栏

leetcode

15 篇文章

订阅专栏

本文介绍了一种高效求解 Top K 频繁元素的问题，通过哈希表统计元素频率，并采用桶排序的方法将不同频率的元素归类到对应的桶中，最终返回频率最高的 K 个元素。该算法的时间复杂度为 O(n)。

347. Top K Frequent Elements

思路：
先建立hash表统计出现的次数. 然后使用桶排序, 按照次数丢入桶中. 时间复杂度O(n)

code:

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        map<int, int> hash;
        for (int i = 0; i < nums.size(); i++)
            hash[nums[i]]++;

        vector<list<int>> buckets(nums.size() + 1);
        for (map<int, int>::iterator it = hash.begin(); it != hash.end(); ++it)  // replace to auto
            buckets[it->second].push_back(it->first);

        vector<int> result;
        for (int i = nums.size(); i >= 0 && result.size() < k; i--)
            //for(list<int>::iterator it = buckets[i].begin(); it != buckets[i].end() ; it++) {
            for (auto it : buckets[i]) {  // STL auto since C++11
                result.push_back(it);
                if (result.size() == k)
                    break;
            }
        return result;
    }
};