Python 二分搜索详解 - 目标值和左右边界 统一模板

来源于 labuladong的算法小抄 

给定 升序 序列，查找3种情况：
1. 查找一个值
2. 查找左侧边界
3. 查找右侧边界

统一套路，使用两端都闭的「搜索区间」，只需要注意 nums[mid] == target条件处的代码 和 最后判断左右索引是否越界

class Solution:
    def binary_search(self, nums, target):
        """
        找不到 target 返回 -1
        :param nums:
        :param target:
        :return:
        """
        if not nums:
            return -1

        left = 0
        right = len(nums) - 1

        # 搜索区间 左右闭
        while left <= right:
            mid = left + (right - left) // 2  # 地板除，向下取整
            if nums[mid] > target:
                # mid 大 => 向左侧区间查找
                right = mid - 1
            elif nums[mid] < target:
                # mid 小 => 向右侧区间查找
                left = mid + 1
            elif nums[mid] == target:
                # 找到 target 直接返回
                return mid
        # 搜索区间为空，没有target
        return -1

    def search_left_bound(self, nums, target):
        if not nums:
            return -1
        left = 0
        right = len(nums) - 1

        while left <= right:
            mid = left + (right - left) // 2
            if nums[mid] > target:
                # mid大 => 向左侧区间查找
                right = mid - 1
            elif nums[mid] < target:
                # mid小 => 向右侧区间查找
                left = mid + 1
            elif nums[mid] == target:
                # 此时说明，左侧边界只能小于等于mid => 左侧区间
                right = mid - 1

        # 最后，判断left是否越界，如果 target 比所有数都大，l 会达到nums的size
        if left >= len(nums) or nums[left] != target:
            return -1
        return left

    def search_right_bound(self, nums, target):
        if not nums:
            return -1
        left = 0
        right = len(nums) - 1

        while left <= right:
            mid = left + (right - left) // 2
            if nums[mid] > target:
                right = mid - 1
            elif nums[mid] < target:
                left = mid + 1
            elif nums[mid] == target:
                # 此时说明，右侧边界只能大于等于mid => 右侧区间
                left = mid + 1

        # 检查 right 是否越界
        if right < 0 or nums[right] != target:
            return -1
        return right