逻辑之旅

  博客园 :: 首页 :: 博问 :: 闪存 :: 新随笔 :: 联系 :: 订阅 订阅 :: 管理 ::
  8 随笔 :: 0 文章 :: 0 评论 :: 0 引用

针对问题

找到一对无序的数中第  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

 

posted on 2019-05-31 00:45  LinMiaoj  阅读(...)  评论(...编辑  收藏