Leetcode 347. 前 K 个高频元素

最新推荐文章于 2025-11-16 01:08:45 发布

原创最新推荐文章于 2025-11-16 01:08:45 发布 · 70 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#C++ #优先队列

LeetCode 专栏收录该内容

210 篇文章

订阅专栏

本文介绍了一种高效算法，用于找出数组中出现频率最高的前 K 个元素。通过使用哈希表统计每个元素的出现次数，并结合优先队列进行筛选，确保了算法的时间复杂度优于 O(nlogn)。此方法适用于数据科学与算法优化等领域。

题目描述

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

示例 1:

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

输入: nums = [1], k = 1
输出: [1]
提示：

你可以假设给定的 k 总是合理的，且 1 ≤ k ≤ 数组中不相同的元素的个数。
你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。
题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的。
你可以按任意顺序返回答案

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/top-k-frequent-elements
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

C++

class Solution {
public:
/*
初步思路：遍历一次存入哈希
然后利用优先队列， pair类型放入优先队列中总是先比较first的大小,相同在比较second的大小
*/
    vector<int> topKFrequent(vector<int>& nums, int k) {
            vector<int>  res;
            unordered_map <int,int>  map;  
            priority_queue<pair<int,int> > pq;  //优先队列  //默认是大顶堆

            for(int i=0;i<nums.size();i++){
                 map[nums[i]]++;
            }
         //始终保持队列中K个元素
            for(const auto&  x:map){
                pq.push(pair<int,int>(-x.second,x.first));
                if(pq.size()>k) pq.pop();  //因为队列删除的时候是从队头删，所以前边用的负值，原来的值越大越靠队尾，就不被删除
            }
           while(k--){
               res.push_back(pq.top().second);
               pq.pop();
           }
           return res;

    }
};