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