最小的k个数

Posted 2021-02-21 huoguonaicha

题目：最小的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;
    }

};