Day17：剑指 Offer 40. 最小的k个数

题目

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

代码实现

快排

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        if(k >= arr.length){
            return arr;
        }
        quickSort(arr,0,arr.length-1);
        int[] res = new int[k];
        for (int i = 0; i < res.length; i++) {
            res[i] = arr[i];
        }
        return res;
    }

    /**
     * 左闭右闭
     * @param arr
     * @param begin
     * @param end
     */
    public void quickSort(int[] arr,int begin,int end){
        if (begin < end){
            //par的左边一定小于等于它，右边一定大于等于它
            int par = partition(arr,begin,end);
            quickSort(arr,begin,par - 1);
            quickSort(arr,par + 1,end);
        }
    }
    public int partition(int[] arr,int begin,int end){
        //以最后一个元素作为划分值
        int index = end;
        //从左开始遍历
        while (begin < end){
            if (arr[begin] <= arr[index]){
                begin++;
            }else {
                end--;
                swap(arr,begin,end);
            }
        }
        //将划分值和左边界值交换
        swap(arr,index,begin);
        return begin;
    }
    public void swap(int[]arr,int i,int j){
        int temp = arr[i];
        arr[i] = arr[j];
        arr[j] = temp;
    }
}

快排优化版

题目只要求返回最小的 k 个数，对这k个数的顺序并没有要求。因此，只需要将数组划分为 最小的 k个数 和 其他数字 两部分即可，而快速排序的哨兵划分可完成此目标。

根据快速排序原理，如果某次哨兵划分后 基准数正好是第 k+1 小的数字 ，那么此时基准数左边的所有数字便是题目所求的 最小的 k个数 。

根据此思路，考虑在每次哨兵划分后，判断基准数在数组中的索引是否等于 k，若 truetrue 则直接返回此时数组的前 k 个数字即可。

K神代码：
以最左边的数为划分

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        if (k >= arr.length) return arr;
        return quickSort(arr, k, 0, arr.length - 1);
    }
    private int[] quickSort(int[] arr, int k, int l, int r) {
        int i = l, j = r;
        while (i < j) {
            while (i < j && arr[j] >= arr[l]) j--;
            while (i < j && arr[i] <= arr[l]) i++;
            swap(arr, i, j);
        }
        swap(arr, i, l);
        if (i > k) return quickSort(arr, k, l, i - 1);
        if (i < k) return quickSort(arr, k, i + 1, r);
        return Arrays.copyOf(arr, k);
    }
    private void swap(int[] arr, int i, int j) {
        int tmp = arr[i];
        arr[i] = arr[j];
        arr[j] = tmp;
    }
}