LeetCode:347. Top K Frequent Elements
Given a non-empty array of integers, return the k most frequent elements.
For example,
Given [1,1,1,2,2,3] and k = 2, return [1,2].
Note:
You may assume k is always valid, 1 ≤ k ≤ number of unique elements.
Your algorithm’s time complexity must be better than O(n log n), where n is the array’s size.
今天我真是重新见识了一下STL,有一堆容器还没有用过。
最下面用vector的是我写的,当然这种写法也是当年学的别人的,现在快成自己的了–3。
第一个没有注释掉的是我参考了别人写的第二种之后,突发奇想写的。用可以重复,默认升序的map装关键字是出现次数,对应内容数字–1。
中间的是priority_queue,先来个相关说明:
priority_queue:
http://www.cnblogs.com/mfryf/archive/2012/09/05/2671883.html
代码参考:
http://blog.csdn.net/ebowtang/article/details/51317106
默认最大堆。以pair为元素类型,还可以默认排序,估计是key排序。注意在获取最大值后要pop()。
class Solution {
public:
// static bool compare(pair<int,int> &pa,pair<int,int> &pb)
// {
// return pa.second>pb.second;
// }
vector<int> topKFrequent(vector<int>& nums, int k) {
unordered_map<int,int> umNums;
for(const int &im:nums)
{
++umNums[im];
}
//1
multimap<int,int> mRes;
for(const pair<int,int> &pm:umNums)
{
mRes.insert({pm.second,pm.first});
}
vector<int> res;
for(auto iter=mRes.rbegin();iter!=mRes.rend()&&res.size()<k;++iter)
{
res.push_back(iter->second);
}
return res;
//2
// priority_queue<pair<int,int>> pq;
// for(const pair<int,int> &pm:umNums)
// {
// pq.push({pm.second,pm.first});
// }
// vector<int> res(k);
// for(int i=0;i<k;++i)
// {
// res[i]=pq.top().second;//最大堆
// pq.pop();
// }
// return res;
//3
// vector<pair<int,int>> pvNums;//vector自己加个比较函数排序
// for(const pair<int,int> &pm:umNums)
// {
// pvNums.push_back(pm);
// }
// sort(pvNums.begin(),pvNums.end(),compare);
// vector<int> res(k);
// for(int i=0;i<k;++i)
// {
// res[i]=pvNums[i].first;
// }
// return res;
}
};