347. Top K Frequent Elements

最新推荐文章于 2025-07-29 19:56:06 发布

timeshark

最新推荐文章于 2025-07-29 19:56:06 发布

阅读量439

点赞数

CC 4.0 BY-SA版权

分类专栏： # 算法实践文章标签： c++ 堆

本文链接：https://blog.youkuaiyun.com/x_shuck/article/details/51448400

算法实践专栏收录该内容

207 篇文章

订阅专栏

本文介绍了一种高效算法，用于找出数组中出现频率最高的k个元素。通过使用哈希映射记录每个元素的出现次数，并借助优先队列按频率排序元素，从而实现O(n log k)的时间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given a non-empty array of integers, return the k most frequent elements.

For example,
Given [1,1,1,2,2,3] and k = 2, return [1,2].

Note:

You may assume k is always valid, 1 ≤ k ≤ number of unique elements.
Your algorithm's time complexity must be better than O(n log n), where n is the array's size.

题意：返回频繁出现次数排名前k的元素。

思路：建立hashmap记录val-count, 然后建立val-count的优先队列（以count值排序），然后吐出前k个pair对应first值即可。

struct cmp
{
	bool operator() (const pair<int, int> a, const pair<int, int>b) const
	{
		return a.second < b.second;
	}
};

class Solution {
public:
	vector<int> topKFrequent(vector<int>& nums, int k) {
		unordered_map<int, int> mymap;
		for (int i = 0; i < nums.size(); i++){
			if (mymap.find(nums[i]) != mymap.end()){
				mymap[nums[i]]++;
			}
			else{
				mymap[nums[i]] = 1;
			}
		}

		priority_queue<pair<int, int>, vector<pair<int, int>>, cmp> myqueue(mymap.begin(), mymap.end());
		vector<int> res(k, 0);
		for (int i = 0; i < k; i++){
			res[i] = myqueue.top().first;
			myqueue.pop();
		}
		return res;
	}	
};