【算法】Quick Select

针对问题

找到一对无序的数中第 K 大，或者第 K 小的元素，返回该元素的值或者它的 index（index 的情况比较适合这堆数每个都独一无二的情况，不然可能会有多个答案）。

关键思想

拿一个数做参照，其他数通过对比它，来左右放置，得到的结果肯定是这个数在该数组中的排列位置是正确的。(DIvide and Conquer 分治思想)

形象讲解

把所有数字当成一个个球，假设我们要选出第 K 小的那个球。

我们有两个筐：A 和 B。

首先我们随机选择一个球作为参照物，然后我们开始分捡这一堆球：比它小的进入 A 筐，大于等于它的进入 B 筐。

然后我们数一下 A 筐中有几个球。

如果刚好有 K-1 个，那么我们的参照物小球就是我们想要的那个第 K 个小球，不用忙活了，直接返回参照球；

如果小于 K-1 个，那么我们要的小球明显在 B 筐里面；我们排除掉 A 筐里面的 a 个球，则我们需要的是 B 筐中第 K-a 小的球，把筐清空，重新选择参照球，按上面流程再来一遍；

如果大于 K-1 个，那么我们要的小球明显在 A 筐里面；我们排除掉 B 筐里面的 b 个球，则我们需要的是 A 筐中第 K 小的球，把筐清空，重新选择参照球，按上面流程再来一遍。

实际与概念的区别

实际写代码的时候的操作，跟上面图解还是有些区别的。

上面的图解是为了简单粗暴快速将关键的概念融会贯通，而实际代码中，因为我们操作的是数组，而且为了防止递归迭代过程中无限循环，所以做了一些改进：

1. 不需要两个筐，在原数组上，选择最后一个元素作为参照数，浏览整个数组（除了参照数），然后在原数组上直接交换元素位置；

2. 直接操作数组的 start_index 和 end_index 就好，无需操作 k ；

3. 递归迭代过程，与二分搜索的思路相似，为了防止出现递归下一层的数的数量与上一层相同，导致无限循环的情况，我们在下一层递归中，会将参照数拿走（即不会把参照小球放入 B 筐）。

JavaScript 代码

function quickSelect(nums, start, end, k){ //注意：k 是 zero-based 的，从零开始计数，而不是从一开始

   if(start > end) return;

    let pivot = nums[end];

    let left = start;

    for(let i=start; i<end; i++){

        if(nums[i] < pivot){ //根据实际需要用 > 或 <

            [nums[i], nums[left]] = [nums[left], nums[i]];

            left++;

        }

    }

    [nums[left], nums[end]] = [nums[end], nums[left]];

    if(left === k) return pivot;

    if(left < k) return quickSelect(nums, left+1, end, k);

    return quickSelect(nums, start, left-1, k);

}

做题链接

LeetCode 215. Kth Largest Element in an Array

巴特西