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) ,要用引用。

相关推荐
仰泳的熊猫9 分钟前
LeetCode:268. 丢失的数字
数据结构·c++·算法·leetcode
VT.馒头9 分钟前
【力扣】2725. 间隔取消
javascript·leetcode·1024程序员节
小龙报25 分钟前
《算法通关指南数据结构和算法篇(3)--- 栈和stack》
开发语言·数据结构·c++·算法·创业创新·学习方法·visual studio
CoovallyAIHub1 小时前
智能“下沉”:边缘AI,更低功耗、更快响应、更强隐私,YOLO26只是开始
深度学习·算法·计算机视觉
遇印记1 小时前
2026蓝桥杯
职场和发展·蓝桥杯
Lear2 小时前
【链表】LeetCode 142.环形链表
算法
CoovallyAIHub2 小时前
2025目标检测模型全景图:从RF-DETR到YOLOv12,谁主沉浮?
深度学习·算法·计算机视觉
澪吟2 小时前
算法性能的核心度量:时间复杂度与空间复杂度全解析
数据结构·算法
咪咪渝粮2 小时前
108. 将有序数组转换为二叉搜索树
算法·leetcode
lzptouch2 小时前
蚁群(Ant Colony Optimization, ACO)算法
人工智能·算法·机器学习