LeetCode刷题——347. 前 K 个高频元素

✊✊✊🌈大家好!本篇文章将较详细介绍栈的题目347.前 K 个高频元素,主要记录小顶堆的使用方式,代码语言为:C++代码😇。


🎡导航小助手🎡

347. 前 K 个高频元素

🔒1、题目:

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

🌲 示例 1🌲:

复制代码
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]

🌲 示例 2🌲:

复制代码
输入: nums = [1], k = 1
输出: [1]

❗️ 提示 ❗️ :

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

来源:力扣(LeetCode)👈

☀️2、思路:

1.遍历nums数组,哈希表mp记录每个数字出现的次数

2.利用queue先进优先小顶堆q,遍历mp,得到代表mp中前k大的数字的数组

3.倒序取出前k个频率最大的数存入vector数组

复杂度分析

⏳时间复杂度 O(NlogK)

🏠空间复杂度 O(N)

🔑3、代码:

cpp 复制代码
class Solution {
public:
	    //1.
	    // class mycomparison {
	    // public:
	    //     bool operator()(const pair<int, int>& lhs, const pair<int, int>& rhs) {
	    //         return lhs.second > rhs.second;
	    //     }
	    // };
	    //2.
	    static bool cmp(pair<int,int>&a,pair<int,int>&b){
		        return a.second > b.second;
	    }
	
	    vector<int> topKFrequent(vector<int>& nums, int k) {
	        vector<int> ans(k);
	        unordered_map<int,int> mp;
	        for(auto num:nums) mp[num]++;//哈希表记录元素出现的次数
	
	        //大小为k的小顶堆
	        //1.
	        //priority_queue<pair<int, int>, vector<pair<int, int>>, mycomparison> pri_q;
	        //2.
	        priority_queue<pair<int, int>, vector<pair<int,int>>,decltype(&cmp)>pri_q(cmp); 
	
	        // 用固定大小为k的小顶堆,扫面所有频率的数值
	        for (auto it = mp.begin(); it != mp.end(); it++) {
	            pri_q.push(*it);
	            if (pri_q.size() > k) { // 如果堆的大小大于了K,则队列弹出,保证堆的大小一直为k
	                pri_q.pop();
	            }
	        }
	        //倒序取出前K个频率最大的数
	        for(int i = k - 1; i >= 0; i--){
	            ans[i]=pri_q.top().first;
	            pri_q.pop();
	        }
	        return ans;
	    }
	};

定义小顶堆的两种方式 :

1.类

cpp 复制代码
	class mycomparison {
	public:
		bool operator()(const pair<int, int>& lhs, const pair<int, int>& rhs) {
			return lhs.second > rhs.second;
		}
	};
	
	priority_queue<pair<int, int>, vector<pair<int, int>>, mycomparison> pri_q;

2.静态函数

cpp 复制代码
	static bool cmp(pair<int,int>&a,pair<int,int>&b){
		return a.second > b.second;
	}
	priority_queue<pair<int, int>, vector<pair<int,int>>,decltype(&cmp)>pri_q(cmp); 

注意decltype(&cmp) ,要用引用。

相关推荐
CoovallyAIHub4 小时前
港大&字节重磅发布DanceGRPO:突破视觉生成RLHF瓶颈,多项任务性能提升超180%!
深度学习·算法·计算机视觉
CoovallyAIHub5 小时前
英伟达ViPE重磅发布!解决3D感知难题,SLAM+深度学习完美融合(附带数据集下载地址)
深度学习·算法·计算机视觉
聚客AI1 天前
🙋‍♀️Transformer训练与推理全流程:从输入处理到输出生成
人工智能·算法·llm
大怪v1 天前
前端:人工智能?我也会啊!来个花活,😎😎😎“自动驾驶”整起!
前端·javascript·算法
惯导马工1 天前
【论文导读】ORB-SLAM3:An Accurate Open-Source Library for Visual, Visual-Inertial and
深度学习·算法
骑自行车的码农1 天前
【React用到的一些算法】游标和栈
算法·react.js
博笙困了1 天前
AcWing学习——双指针算法
c++·算法
moonlifesudo1 天前
322:零钱兑换(三种方法)
算法
NAGNIP2 天前
大模型框架性能优化策略:延迟、吞吐量与成本权衡
算法
美团技术团队2 天前
LongCat-Flash:如何使用 SGLang 部署美团 Agentic 模型
人工智能·算法