码迷,mamicode.com
首页 > 其他好文 > 详细

最小的k个数

时间:2020-05-01 01:17:17      阅读:110      评论:0      收藏:0      [点我收藏+]

标签:review   art   超过   attach   res   返回   bsp   tab   -name   

题目:最小的k个数

输入n个整数,找出其中最小的k个数。

注意:

  • 数据保证k一定小于等于输入数组的长度;
  • 输出数组内元素请按从小到大顺序排序;

样例:

输入:[1,2,3,4,5,6,7,8] , k=4

输出:[1,2,3,4]

思路:

  用大根堆来解决此题。大根堆的特征,大根堆的堆顶元素一定是最大的元素。

  遍历数组,将每个元素依次插入大根堆,每插入一个数判断一下当前大根堆里元素的个数是否超过k个数,如果超过了k个数,则弹出堆顶元素。

  时间复杂度:大根堆插入一个元素为logk,n个元素每个元素进行插入到大根堆中,所以时间复杂度为o(nlogk)

代码:

class Solution {
public:
    vector<int> getLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> res;
        priority_queue<int> heap;
        for(auto x : input){

            //将每个元素依次插入大根堆
            heap.push(x);

            //判断一下当前大根堆里元素的个数是否超过k个数,如果超过了k个数就弹出最大的值
            if(heap.size() > k) {
                heap.pop();
            }
        } 
        for(int i = 0;i < k;i++){

            //res[i] = heap.pop();
            //会报错error: void value not ignored as it ought to be
            //原因将一个返回值的函数赋值给了一个变量.
            //res.push_back ( heap.pop() );
            //报错:invalid use of void expression
            //函数的类型为最后vector<int> ,所以将大根堆的数写回到数组中。
            res.push_back ( heap.top());
            heap.pop();
        }

         //res数组此时为从大到小,所以要翻转数组。
         reverse(res.begin(),res.end());
         return res;
    }

};

 



最小的k个数

标签:review   art   超过   attach   res   返回   bsp   tab   -name   

原文地址:https://www.cnblogs.com/huoguonaicha/p/12811842.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!