利用优先级队列求频率最高的k个元素

leetcode 347. 前 K 个高频元素

给定一个非空的整数数组，返回其中出现频率前 k 高的元素。
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:
输入: nums = [1], k = 1
输出: [1]

思路

1.先用哈希map统计每个元素出现的次数
2.再用优先级队列priority_queue记录频率最大的k个数
3.在将队列中的元素输出到结果集vector中

//优先级队列默认是最大堆，元素从小到大输出，这里采用自定义的比较函数，利用最小堆的方式，队列只保留最大的k个数
class cmp{
	public:
	bool opreator()(const pair<int, int> item1, const pair<int, int> item2){
		return item1.second > item2.second;
	}
};
vector<int> topKFrequent(vector<int>& nums, int k){
	//记录元素的频率
	unordered_map<int, int> numsMap;
	for(auto n : nums)
		++numsMap[n];
	//优先级队列
	priority_queue<pair<int, int>, vector<pair<int,int>>, cmp> pri_que;
	for(auto item : numsMap){
		if(pri_que.size() == k){
			pri_que.push(item);
			pri_que.pop();
			continue;
		}
		pri_que.push(item);
	}
	vector<int> result(k);
	while(--k >= 0){
		result[k] = pri_que.top.first;
		pri_que.pop();
	}
	return result;
}