LeetCode 347.前k个高频元素【c++】

最新推荐文章于 2025-02-04 23:43:27 发布

原创最新推荐文章于 2025-02-04 23:43:27 发布 · 1k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #c++ #算法

LeetCode 专栏收录该内容

22 篇文章

订阅专栏

本文介绍了一种高效的算法，用于从整数数组中找出出现频率最高的前k个元素。该方法利用unordered_map统计频数，结合priority_queue进行排序，最终实现优于O(nlogn)的时间复杂度。

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

一、题目

给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按任意顺序返回答案。

示例 1:

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

示例 2:

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

提示：

1 <= nums.length <= 10的5次方 k 的取值范围是 [1, 数组中不相同的元素的个数]
题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的

进阶：

你所设计算法的时间复杂度必须优于 O(n log n) ，其中 n 是数组大小。

二、解题思路

我是利用堆来解题的。
先利用 unordered_map 计算出每个元素的频数，然后再利用优先队列 priority_queue 来排序，最后再输出符合题意的值

三、代码

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> mp;      //先计算每个元素的频数
        int length=nums.size();
        for(int i=0;i<length;i++){
            mp[nums[i]]++;
        }
        priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>> > heap;       //利用 优先队列 进行排序
        for(auto j:mp){
            heap.push(make_pair(j.second,j.first));
            if(heap.size()>k){
                heap.pop();
            }
        }
        vector<int> arr;            //将符合题意的元素放入数组arr中，并返回结果
        for(int i=0;i<k;i++){
            arr.push_back(heap.top().second);
            heap.pop();
        }
        return arr;
    }   
};

执行用时：8 ms, 在所有 C++ 提交中击败了97.21%的用户
内存消耗：13.4 MB, 在所有 C++ 提交中击败了38.83%的用户

四、补充知识

1.unordered_map 与 map

不同之处：
（1）需要引入的头文件不同
unordered_map: #include <unordered_map>
map: #include < map >
(2) 内部实现机理不同
unordered_map:unordered_map内部实现了哈希表。元素排列无序
map:map内部实现了红黑树。元素排列有序

它俩用法一样
里面的元素以pair类型来存储。
pair
（1）pair是将两个数据组合成一组数据，有这样的需求时可以使用pair。STL中的map就是将key、value放一起。
（2）pair中俩元素可通过first、second访问。
（3）利用 make_pair 可创建新的pair对象。

2.priority_queue

priority_queue即优先队列，其本质是一个堆实现的。它和队列基本操作相同。
定义：
priority_queue<Type,Container,Functional>
Type:数据类型
Container:容器类型。必须是用数组实现的容器，eg.vector
Functional：比较方式。默认是大根堆。
eg.

priority_queue<int,vector<int>,greater<int> > heap;		//升序
priority_queue<int,vector<int>,less<int> > heap;		//降序

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/top-k-frequent-elements