LeetCode347：前K个高频元素

本文主要是介绍LeetCode347：前K个高频元素，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

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

在这里插入图片描述
解题思想
使用优先队列
priority_queue<Type, Container, Functional>
Type 就是数据类型，Container 就是容器类型（Container必须是用数组实现的容器，比如vector,deque等等，但不能用 list。STL里面默认用的是vector），Functional 就是比较的方式，当需要用自定义的数据类型时才需要传入这三个参数，使用基本数据类型时，只需要传入数据类型，默认是大顶堆

//升序队列
priority_queue <int,vector<int>,greater<int> > q;
//降序队列
priority_queue <int,vector<int>,less<int> >q;//greater和less是std实现的两个仿函数
//（就是使一个类的使用看上去像一个函数。其实现就是类中实现一个operator()，这个类就有了类似函数的行为，就是一个仿函数类了

代码

class Solution {
public:class cmp{public:bool operator()(const pair<int,int>& l, const pair<int,int>& r){return l.second > r.second;}};vector<int> topKFrequent(vector<int>& nums, int k) {unordered_map<int,int> umap;for(int i = 0; i < nums.size(); ++i){++umap[nums[i]];}priority_queue<pair<int, int>, vector<pair<int, int>>, cmp> pri_que;for(auto it = umap.begin(); it != umap.end(); ++it){pri_que.push(*it);if(pri_que.size() > k) pri_que.pop();}vector<int> res(k);for(int i = 0; i < k; i++){res[i] = pri_que.top().first;pri_que.pop();}return res;}
};