二分查找 红蓝染色法 - 灵神视频总结

这节课的内容主要讲了二分查找，基础知识。

问题：返回有序数组第一个>= 8 的数据的位置，如果所有的数都小于8，则返回数组的长度。

暴力的做法： 一次遍历，找到第一个>=8的位置。 时间复杂度 O(n)

如何优化？ 二分查找。前提条件是数组是有序的。

设置两个指针 L R ，分别指向数组的第0个元素和第n-1个元素。 并从中选择一个元素M，去对比和target的关系。通常我们使用 M = (L +R)//2, 向下取整数。

我们约定

红色背景代表false，即 < target 。

蓝色背景代表true， 即 >= target。

白色背景代表不确定。

在题中，我们发现 nums[M] < 8, 那么它左边的元素都要涂成红色的。

这时候我们确定了红色部分，不确定的部分只有右半部分[M+1, R]了， 需要把 L <- M+1。

🚩 重点说明： L-1 一定是红色的

下一步 继续轮询，意思就是说：继续给数组涂色。

这时候发现nums[M] >= target ，满足条件了，那么就给[M,R] 都涂上蓝色。

然后让R = M-1

🚩 重点说明 R+1 一定是蓝色的

继续轮询，

发现nums[M] >= target 满足条件了，要涂成红色，这时候让R = M-1 。

至此，所有的数组都已经知道了。

给这个东西起个名字： 循环不变量 L-1 左边的元素都是红色， R+1 右边的元素都是蓝色。

https://juejin.cn/post/6876426140899082253

循环不变量这个好难理解，我的理解也是这样，就是L R的边界的定义一直不变。

回顾我们的问题是什么呢
 

问题：返回有序数组第一个>= 8 的数据的位置，如果所有的数都小于8，则返回数组的长度。

那答案显而易见了。

第一个位置就是R+1 ，由于L = R+1 ，直接返回L即可。

问题2: 如果所有的元素都比target 小呢，

答案： 这时候我们可以想一下 L 和 R 分别在哪里， R的位置不变， L的位置会移动到数组外面的位置上去。

那么 L 就等于数组的长度。

代码 ：

这里给定了三种解法，本人觉得只需要明白一种，养成习惯只写一种就好了，不然很容易乱。

# 返回有序数组第一个>= 8 的数据的位置，如果所有的数都小于8，则返回数组的长度。
def lower_bound(nums, target):
    n = len(nums)
    left = 0
    right = n -1 
    while left <= right : #区间不为空。
        mid = (left + right) //2 
        if nums[mid] < target: # [mid+1, right]
            left = mid + 1
        else : # [left, mid-1]
            right = mid - 1
        
    return left 

def lower_bound2(nums, target):
    n = len(nums)
    left = 0
    right = n 
    while left < right : #区间不为空。
        mid = (left + right) //2 
        if nums[mid] < target: # [mid+1, right)
            left = mid + 1
        else : # [left, mid)
            right = mid 
        
    return left 



def lower_bound3(nums, target):
    n = len(nums)
    left = -1
    right = n 
    while left + 1 < right : #区间不为空。
        mid = (left + right) //2 
        if nums[mid] < target: # (mid, right)
            left = mid 
        else : # (left, mid)
            right = mid 
    return right  

问题： 如果把上题改了，不是>= target了，而是变成了 > , <, <= target呢，阁下该如何处理呢？

我们可以把有序数组上的二分查找就分为这四种形式。

🚩

1. 第一个 >=x 的下标 ： lower_bound(nums, x)
2. 第一个 > x 的下标=> 第一个大于等于x+1 的下标。lower_bound(nums, x +1 )
3. 最后一个一个小于x的下标：可以转换为数组中第一个大于等于 x 的下标 的左边位置, low_bound(x) - 1;
4. 最后一个小于等于x的下标：可以转换为数组中第一个大于等于 x+1 的下标 的 左边位置, low_bound(x+1) - 1;

34. 在排序数组中查找元素的第一个和最后一个位置

直接给出代码：

def lower_bound(nums, target):
    n = len(nums)
    left = 0
    right = n -1 
    while left <= right : #区间不为空。
        mid = (left + right) //2 
        if nums[mid] < target: # [mid+1, right]
            left = mid + 1
        else : # [left, mid-1]
            right = mid - 1
        
    return left 
class Solution(object):
    def searchRange(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        left = lower_bound(nums, target)
        print(left)
        if left == len(nums) or nums[left] != target:
            return [-1, -1]
        end = lower_bound(nums, target+1) - 1
        print(end)

        return [left, end ]