leetcode.347 前 K 个高频元素

347. 前 K 个高频元素

难度中等449收藏分享切换为英文关注反馈

给定一个非空的整数数组，返回其中出现频率前 k 高的元素。

 

示例 1:

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

示例 2:

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

 

提示：

• 你可以假设给定的 k 总是合理的，且 1 ≤ k ≤ 数组中不相同的元素的个数。
• 你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。
• 题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的。
• 你可以按任意顺序返回答案。

用优先队列模拟小跟堆

优先队列是默认最大的一直在最前面

这样的话先建立pair<int, int>的优先队列

确认每一个数字的次数

按照次数大小建立优先队列

当优先队列大小大于k时，就推出

这时候推出的是最前面的，也就是最大的

而我们应该推出最小的，所以这里用次数的负数做为标的建立优先队列

#include <iostream>
#include <unordered_map>
#include <queue>


using namespace std;

vector<int> topKFrequent(vector<int>& nums, int k) {
	vector<int> re;
	unordered_map<int, int> mp;
	priority_queue< pair<int, int> > pq;
	int i;
	for(i = 0; i < nums.size(); i++){
		mp[nums[i]]++;
	}
	for (auto p : mp) {
        pq.push(pair<int, int>(-p.second, p.first));
        if (pq.size() > k) pq.pop();
    }

    while (k--) {
	    re.push_back(pq.top().second);
	    pq.pop();
	}
	return re;
	


}