LeetCode：347. Top K Frequent Elements

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;
}
};