Python3-[43]面试题53-I.在排序数组中查找数字I（二分查找）

给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
你的算法时间复杂度必须是 O(log n) 级别。
如果数组中不存在目标值，返回 [-1, -1]。
示例 1:

输入: nums = [5,7,7,8,8,10], target = 8
输出: [3,4]

示例 2:

输入: nums = [5,7,7,8,8,10], target = 6
输出: [-1,-1]

P.S.题目来源于leetcode。

二分查找

两区间法：
704/33/81/34/35/4/153/154/300/275/1095 查找元素索引
69/287/374 在有界区间内搜索整数
278/410/658/875/1300 判断条件为函数

mid在左区间：

while low < high: 
    mid = low + [(high - low ) / 2]  
    if target in [low, mid ]:
        high = mid 
    elif target in [mid + 1，high]:
        low = mid + 1

mid在右区间

while low < high: 
    mid = low + [(high - low + 1) / 2]  ##mid在右区间时需要上取整
    if target in [low, mid - 1]:
        high = mid - 1
    elif target in [mid，high]:
        low = mid

[low, mid], [mid + 1, high]: 返回目标左索引或大于目标的第一个元素的索引。
[low, mid - 1], [mid, high]: 返回目标右索引或小于目标的第一个元素的索引.

解题思路

二分查找：找到满足条件的左边界/右边界。
将整个代搜索区间划为2部分：一定不包含target的区间和可能包含target的区间。可以按照mid划归为左区间或右区间分为：[low, mid]和[mid+1, high] 以及 [low, mid-1]和[mid, high]。对于mid 划归为右区间的情形，注意求mid时需要为上取整，否则mid 达不到high值，陷入死循环。
找左边界=>把mid划归到左区间。
找右边界=>把mid划归到右区间。
while low < high:可以满足到循环结束时low == high。
mid划归到右区间时low = mid, 注意这时mid求解是采用了上取整。mid = (low + high + 1) // 2或者 mid = low + (high - low + 1) // 2.

代码

#给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。时间复杂度：O(nlogn),如果数组中不存在目标值，返回 [-1, -1]。
class Solution:
    def searchRange(self, nums: List[int], target: int) -> List[int]:
        #顺序数组，要求O(nlogn)的时间复杂度，应该使用二分查找
        
        def binsear_right(nums,tar):#找右边界，mid在右区间
            low, high = 0, len(nums) - 1
            while low < high:#最后返回时会有low == high.
                mid = (low + high + 1) // 2#上取整 
                if nums[mid] <= tar:
                    low = mid 
                else:
                    high = mid - 1
            if nums[low] == tar:
                return low
            else:return -1

        def binsear_left(nums,tar):#找左边界，mid在左区间
            low, high = 0, len(nums) - 1
            while low < high:
                mid = (low + high) // 2#下取整
                if nums[mid] >= tar:
                    high = mid
                else:
                    low = mid + 1
            if nums[low] == tar:
                return low
            else:return -1

        if len(nums) == 0: return [-1, -1]
        left_idx = binsear_left(nums, target)
        if left_idx == -1:return [-1, -1]#没有左边界亦无右边界。
        right_idx = binsear_right(nums, target)
        return [left_idx, right_idx]