LeetCode刷题046 最小的k个数

剑指 Offer 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]

限制：

0 <= k <= arr.length <= 10000
0 <= arr[i] <= 10000

题目

快速排序

Code

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

    public void quickSort(int[] arr, int a, int b){
        if(a >= b){
            return;
        }
        int l = a - 1, r = b + 1;
        int mid = arr[ (a + b) >> 1];
        while(l < r){
            do l++; while(arr[l] < mid);
            do r--; while(arr[r] > mid);
            if(l < r){
                int tmp = arr[l];
                arr[l] = arr[r];
                arr[r] = tmp;
            }
        }
        quickSort(arr, a, r);
        quickSort(arr, r + 1, b);
    }
}