347.前K個高頻元素（Leecode刷題）

給定一個非空的整數數組，返回其中出現頻率前 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可以。

思路很簡單，但是我在寫的過程中發現了幾個問題：

1. 我想着是用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;

    }