当前位置:   article > 正文

347.前K个高频元素(Leecode刷题)_整理出 reviews 中出现频率最高的前 k 种语言 leecode

整理出 reviews 中出现频率最高的前 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;

    }
    
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/你好赵伟/article/detail/677772
推荐阅读
相关标签
  

闽ICP备14008679号