赞
踩
给定一个非空的整数数组,返回其中出现频率前 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可以。
思路很简单,但是我在写的过程中发现了几个问题:
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; }
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。