【算法】Quick Select

针对问题

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

 

关键思想

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

 

形象讲解

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

 

我们有两个筐:AB

 

首先我们随机选择一个球作为参照物,然后我们开始分捡这一堆球:比它小的进入 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);
}

 

相关概念

pivot:轴,从无序数组中随机选取一个数字,或者取它的 first element 或 last element。

 

做题链接

LeetCode 215. Kth Largest Element in an Array

 

转载于:https://www.cnblogs.com/LinMiaoJia/p/QuickSelect.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值