面试题40. 最小的k个数（堆排序）

面试题40. 最小的k个数

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

 

示例 1：

输入：arr = [3,2,1], k = 2
输出：[1,2] 或者 [2,1]

示例 2：

输入：arr = [0,1,2,1], k = 1
输出：[0]

思路：用库函数做就没意思了，这道题我主要用来复习一下手撕堆排序（算法笔记上的）

完整代码：

class Solution {
public:
    //调整位置函数，让下标为low的元素找到自己的位置
    void downAdjust(int low,int high,vector<int>& arr){
        int i = low;
        int j = i*2+1;
        while(j<high){
            if(j+1<high&&arr[j]<arr[j+1]){
                ++j;
            }
            if(arr[i]<arr[j]){
                swap(arr[i],arr[j]);
                i = j;
                j = i*2+1; //注意下标是从0开始的，所以左孩子为i*2+1，右孩子为i*2+2
            }
            else break;
        }
    }
    //建堆
    void createHeap(vector<int>&arr){
        int size = arr.size();
        for(int i = size/2;i>=0;i--){
            downAdjust(i,size-1,arr);
        }
    }
    //堆排序
    void heapSort(vector<int>&arr){
        createHeap(arr);
        for(int i = arr.size()-1;i>=0;i--){
            swap(arr[0],arr[i]);
            downAdjust(0,i,arr);
        }
    }
    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        heapSort(arr);
        vector<int> res;
        if(arr.size()==0)return res;
        for(int i = 0;i<k;i++){
            res.push_back(arr[i]);
        }
        return res;
    }
};