LeetCode：前 K 个高频元素

最新推荐文章于 2024-10-28 15:00:00 发布

原创最新推荐文章于 2024-10-28 15:00:00 发布 · 305 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #前K个高频元素

LeetCode 专栏收录该内容

29 篇文章

订阅专栏

本文介绍了一种高效的算法，用于从整数数组中找出出现频率最高的k个元素。通过使用哈希表统计每个元素的出现次数，并结合优先队列进行排序，可以快速定位到所需元素。该算法适用于大数据量的场景，能有效提升数据处理效率。

给定一个非空的整数数组，返回其中出现频率前 k 高的元素。

示例 1:

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]

示例 2:

输入: nums = [1], k = 1
输出: [1] 。

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        vector<int> res;
        unordered_map<int, int> M;
        priority_queue<pair<int, int>> Q;
        for(int i =0 ; i < nums.size(); i++)
            M[nums[i]]++;
        for(auto it: M)
            Q.push({it.second, it.first});
        for(int i = 0; i < k; i++)
        {
            int top = Q.top().second;
            Q.pop();
            res.push_back(top);
        }
        return res;
    }
};