剑指 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
基于快速排序的方法
总体思路,先对整个数组进行分区,获得划分分区的那个值的索引index,比较index和所求个数k的大小;
index > k:说明k个最小值都在分区的左部,继续对左部进行partition
index < k:说明k个最小值包含分区的左部所有值,并加上分区右部的部分值,所以继续对右部进行partition
index = k:说明index左部的值刚好k个,即所求的k个最小值
partition思路:(这个耗时较长)
先选择基准值,一般选最后一位,使用两个指针i,j,然后后从0开始遍历;
-[0,i)为小于基准值的数
-[i,j)为为遍历
-[j,end)为大于基准值的数
当前值<=基准值,i指针右移;
当前值>基准值,当前值和j指针所指数对调,j指针左移;
当指针i>j时,表示所有数遍历完成,结束遍历,然后把基准值和i所指的数对调
/**
* @param {number[]} arr
* @param {number} k
* @return {number[]}
*/
var getLeastNumbers = function(arr, k) {
//考虑特殊情况,当所求的数个数大于等于数组的长度,则之间返回数组即可
if(arr.length <= k) return arr;
//获得基准值的索引
let index = partition(arr,0,arr.length-1);
while