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、代码：

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.类

	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.静态函数

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