Leetcode刷题笔记——剑指 Offer 40. 最小的k个数（简单）

---

题目描述

输入整数数组 arr ，找出其中最小的 k 个数。例如，输入4、5、1、6、2、7、3、8这8个数字，则最小的4个数字是1、2、3、4。

方法一：sort排序

这种考查排序方法的题目中直接用sort进行排序虽然简单，但是在面试中不是最佳选择。

方法二：堆

用一个大根堆实时维护数组的前 k 小值。首先将前 k 个数插入大根堆中，随后从第 k+1 个数开始遍历，如果当前遍历到的数比大根堆的堆顶的数要小，就把堆顶的数弹出，再插入当前遍历到的数。最后将大根堆里的数存入数组返回即可。

复杂度分析

时间复杂度：O(nlogk)，其中 n 是数组 arr 的长度。由于大根堆实时维护前 k 小值，所以插入删除都是 O(logk) 的时间复杂度，最坏情况下数组里 n 个数都会插入，所以一共需要 O(nlogk) 的时间复杂度。
空间复杂度：O(k)，因为大根堆里最多 k 个数。

C++代码

class Solution {
public:
    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        vector<int> vec(k, 0);
        if (k == 0) { // 排除 0 的情况
            return vec;
        }
        priority_queue<int> Q;
        for (int i = 0; i < k; ++i) {
            Q.push(arr[i]);
        }
        for (int i = k; i < (int)arr.size(); ++i) {
            if (Q.top() > arr[i]) {
                Q.pop();
                Q.push(arr[i]);
            }
        }
        for (int i = 0; i < k; ++i) {
            vec[i] = Q.top();
            Q.pop();
        }
        return vec;
    }
};

方法三：自建堆

补充：
堆是一种非线性结构，可以把堆看作一个数组，也可以被看作一个完全二叉树，通俗来讲堆其实就是利用完全二叉树的结构来维护的一维数组。
按照堆的特点可以把堆分为大顶堆和小顶堆：
大顶堆：每个结点的值都大于或等于其左右孩子结点的值
小顶堆：每个结点的值都小于或等于其左右孩子结点的值

C++代码

class Solution {
public:
    void adjustHeap(vector<int>& arr, int i, int heapsize){//大顶堆
        int l = 2*i+1, r = 2*i+2, maxindex = i;
        if(l<heapsize && arr[l]>arr[maxindex])
            maxindex = l;
        if(r<heapsize && arr[r]>arr[maxindex])
            maxindex = r;
        if(maxindex != i){
            swap(arr[i], arr[maxindex]);
            adjustHeap(arr, maxindex, heapsize);
        }
    }
    void buildHeap(vector<int>& arr, int heapsize){
        for(int i = (heapsize-1)/2; i>=0; --i){
            adjustHeap(arr, i, heapsize);
        }
    }
    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        vector<int> res;
        if(k==0) return res;
        buildHeap(arr, k);
        for(int i = k; i<arr.size(); ++i){
            if(arr[i] < arr[0]){
                swap(arr[i], arr[0]);
                adjustHeap(arr, 0, k);
            }
        }
        for(int i = 0; i<k; ++i){ 
            res.push_back(arr[i]);
        }
        return res;
    }
};

参考链接

【1】https://leetcode-cn.com/problems/zui-xiao-de-kge-shu-lcof/solution/zui-xiao-de-kge-shu-by-leetcode-solution/
【2】https://www.cnblogs.com/lanhaicode/p/10546257.html