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;

    }
    
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章