前k个高频元素（C++）

题目：给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。

leetcode链接

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

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

思路：

1. 统计每个元素出现次数，我们可以使用unordered_map
2. 然后按出现频率给元素排序，可以考虑最小堆。优先队列是用堆实现的，可以使用优先队列。

代码：

class Solution {	//前K个高频元素
public:
	class mycomparison {	//升序
	public:
		bool operator()(const pair<int, int>& Ihs, const pair<int, int>& rhs) {
			return Ihs.second > rhs.second;
		}
	};
	vector<int> topKFrequent(vector<int>& nums, int k) {
		vector<int> result(k);
		unordered_map<int, int> umap;
		for (int i : nums) {
			umap[i]++;
		}
		
		//定义小顶堆
		priority_queue<pair<int, int>, vector<pair<int, int>>, mycomparison> pri_que;
		unordered_map<int, int>::iterator iter;
		for (iter = umap.begin(); iter != umap.end(); ++iter) {
			pri_que.push(*iter);
			if (pri_que.size() > k) {	//保证堆里元素一直为k个
				pri_que.pop();
			}
		}

		for (int i = k - 1; i >= 0; --i) {
			result[i] = pri_que.top().first;
			pri_que.pop();
		}
		return result;
	}
};

int main() {

	vector<int> nums = { 1, 1, 1, 2, 2, 3 };
	int k = 2;
	Solution s;
	vector<int> result = s.topKFrequent(nums, k);
	for (int i : result){
		cout << i << "\t";
	}
	cout << endl;
	return 0;
}

参考资料：代码随想录