JZJZJZ---最小的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]

代码：

var getLeastNumbers = function(arr, k) {
    var newArr = [];
    quickSort(arr, 0, arr.length - 1);
    for(var i = 0; i < k; i ++) {
        newArr.push(arr[i]);
    }
    return newArr;
};

function quickSort(arr, left, right) {
    var i = left;
    var j = right;
    var key = arr[left];

    if(left >= right) {
        return;
    }

    while(i < j) {
        while(arr[j] > key && i < j) {
            j --;
        }
        while(arr[i] <= key && i < j) {
            i ++;
        }
        if(i < j) {
            var temp = arr[i];
            arr[i] = arr[j];
            arr[j] = temp;
        }
    }

    arr[left] = arr[i];
    arr[i] = key;

    quickSort(arr, left, i - 1);
    quickSort(arr, i + 1, right);
}