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

相关推荐
im_AMBER20 小时前
数据结构 06 线性结构
数据结构·学习·算法
earthzhang20211 天前
【1028】字符菱形
c语言·开发语言·数据结构·c++·算法·青少年编程
papership1 天前
【入门级-算法-3、基础算法:二分法】
数据结构·算法
通信小呆呆1 天前
收发分离多基地雷达椭圆联合定位:原理、算法与误差分析
算法·目标检测·信息与通信·信号处理
丁浩6661 天前
Python机器学习---2.算法:逻辑回归
python·算法·机器学习
伏小白白白1 天前
【论文精度-2】求解车辆路径问题的神经组合优化算法:综合展望(Yubin Xiao,2025)
人工智能·算法·机器学习
无敌最俊朗@1 天前
数组-力扣hot56-合并区间
数据结构·算法·leetcode
囚生CY1 天前
【速写】优化的深度与广度(Adam & Moun)
人工智能·python·算法
码农多耕地呗1 天前
力扣94.二叉树的中序遍历(递归and迭代法)(java)
数据结构·算法·leetcode
微笑尅乐1 天前
BFS 与 DFS——力扣102.二叉树的层序遍历
leetcode·深度优先·宽度优先