Leetcode：剑指offer40：最小的k个数

快排的时间复杂度为O(nlogn)~O(n)，是最快的排序算法，所以找到最小的k个数又不要求顺序的话，快排每次递归以基准值分割是个不缺的想法。
所以可以进行快速排序，当找到基准值的下标是k的时候，说明基准值前面有k个待排序数组中的小的数。
直接把前面的k个数当成子数组输出即可。

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

    public int[] quickSort(int[] arr,int low,int high,int k){
        int middle=findMid(arr,low,high);
        //当基准值的下标与k个数的下标相等时，把分割好的前面k个无序比基准值小的子数组复制并return
        if(middle==k){
            return Arrays.copyOf(arr,k+1);
        }
        return middle>k?quickSort(arr,low,middle-1,k):quickSort(arr,middle+1,high,k);
    }

    public int findMid(int[] arr,int low,int high){
        int flag=arr[low];
        while(low<high){
            while(low<high && arr[high]>flag){
                high--;
            }
            arr[low]=arr[high];
            while(low<high && arr[low]<=flag){
                low++;
            }
            arr[high]=arr[low];
        }
        arr[low]=flag;
        return low;
    }
}

标签：offer40,arr,return,基准值,int,个数,high,low,Leetcode
来源： https://www.cnblogs.com/lisamihama/p/16093349.html