给定一个非空的整数数组,返回其中出现频率前 k 高的元素。
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:
输入: nums = [1], k = 1
输出: [1]
这题的思路是:
1.首先要统计每个数字的次数,unordered_map当仁不让。
2.我们要选出前K大,很熟悉,priority_queue可以。
思路很简单,但是我在写的过程中发现了几个问题:
- 我想着是用lambda来比较,在用lambda的时候发现了几个问题。priority_queue一定要注意,传的模板参数第三个不是传函数对象,而是模板类型,而且如果用的自定义的函数,一定要初始化。
auto funct = [m](int a,int b){return m.at(a) < m.at(b);};
priority_queue<int,vector,decltype(funct)> q(funct);
lambda与priority_queue
注意记得传入你的函数值:
explicit priority_queue( const Compare& compare = Compare(),
const Container& cont = Container() );
2.这里还有一个问题就是lambda的传递一般是const(除非用mutable),所以你捕获m,不能用[]取值,应该用at()。
上述方法超时了!!!我也不知道为啥。难道是自定义的lambda表达式???
后来看了题解,可以不自定义比较函数,直接把计数当做first,然后把数字当做second,传入priority.
代码如下:
vector<int> topKFrequent(vector<int>& nums, int k) {
unordered_map<int,int> m;
for(const int a:nums){
m[a]++;
}
// auto funct = [m](int a,int b){return m.at(a) < m.at(b);};
// priority_queue<int,vector<int>,decltype(funct)> q(funct);
priority_queue<pair<int,int>> q;
for(auto it = m.begin();it != m.end();++it){
q.push(make_pair(it->second,it->first));
}
vector<int> ret;
for(int i = 0 ;i < k;++i){
ret.push_back((q.top()).second);
q.pop();
}
return ret;
}