【leetcode】347. Top K Frequent Elements

最新推荐文章于 2024-10-07 07:09:00 发布

拓荒探险者

最新推荐文章于 2024-10-07 07:09:00 发布

阅读量212

点赞数

分类专栏： # LeetCode 文章标签： leetcode 算法数据结构 c++

本文链接：https://blog.csdn.net/zxc120389574/article/details/106388560

版权

LeetCode 专栏收录该内容

243 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何解决LeetCode上的347题，即在整数数组中找出出现频率最高的前K个元素。作者分享了两种C++解决方案：第一种使用unordered_map和priority_queue，第二种利用unordered_map结合按频率分类的buckets。

摘要由CSDN通过智能技术生成

题目：
Given a non-empty array of integers, return the k most frequent elements.

思路：
抄作业https://leetcode.com/problems/top-k-frequent-elements/discuss/81676/C%2B%2B-O(nlogk)-and-O(n)-solutions。

思路1：用unordered_map对各个元素进行计数，用priority_queue找到第k大的元素，然后从头到尾遍历unordered_map即可出答案。
思路2：用unordered_map对各个元素进行计数，buckets根据不同的出现次数对元素进行分类存放。

代码实现：
思路1：

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int, int> counts;
        priority_queue<int, vector<int>, greater<int>> max_k; // 建立一个小堆，选取最大的k个元素
        
        for (int i = 0; i < nums.size(); ++i){
            ++counts[nums[i]];
        }
        
        for (auto & i : counts){
            max_k.push(i.second);
            while (max_k.size() > k){ // 保证只留下k个最大元素
                max_k.pop();
            }
        }
        
        vector<int> ans;
        for (auto & i : counts){
            if (i.second >= max_k.top()){
                ans.push_back(i.first);
            }
        }
        
        return ans;
    }
};

思路2：

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int, int> counts;
        for (int i = 0; i < nums.size(); ++i){
            ++counts[nums[i]];
        }
        
        vector<vector<int>> buckets(nums.size()+1);
        for (auto & k : counts){
            buckets[k.second].push_back(k.first);
        }
        reverse(buckets.begin(), buckets.end());
        
        vector<int> ans;
        for (auto & bucket : buckets){
            for (auto & e : bucket){
                ans.push_back(e);
                if (ans.size() >= k){
                    return ans;
                }
            }
        }
        
        return ans;
    }
};