这是写二分查找最根本的问题, 因为根据不同的情况, 下面的操作是完全不同的, 这也是很多朋友感觉二分查找难写的地方, 这里为了方便, 我直接将区间定义为左闭右开[left, right)的形式, 这里根据自己的习惯最好将区间完全固定住, 以后都按照一种模式来写, 如果你习惯左闭右闭也同样固定住, 这样不用每次都纠结.

固定为左闭右开的好处是: 很多函数中对区间的操作都是左闭右开的形式, 这算是语言的特点吧, 比如字符串中截取子串长度, 列表的截取都是左闭右开的形式, 这样就可以和语言特点相对应.

另一个好处就是在返回值的问题上, 下文会提到.

当区间固定住为左闭右开, 数组长度就固定为[0, len(nums)), 这也是左右端点的初值, 如为左闭右闭, 那长度就是[0, len(nums)-1]了.

这里也要特别注意, 代码为 while left < right 因为区间是右开, 所以不能有=, 否则区间就不存在了, 这样写循环控制的另外一个好处就是在退出循环时, 必然满足 left == right, 这样在最后的返回值就可以任意返回了, 因为它们完全相等, 而不用去纠结该返回哪个端点了.

写法为 mid = left + (right - left)//2 , //2在Python中表示整除, /2在Python中表示正常除, 是有余数的, 这点和其他语言有所不同. 这样写的好处也非常简单, 就是为了防止大数溢出, 因为写成(left + right)//2时, 当数比较大时, left + right是有溢出的风险的, 这种写法就可以避免了.

if nums[mid] < target  这个比较关系涉及到上图中4和5的两种变形, 写成什么样的形式是不固定的, 这里是根据题目要求来写的. 

简单来说就是如果是<, 那么当nums[mid]达到是比target小的数中的最大的数的时候, 通过if条件进入, left的值为mid+1后的值必然是大于等于target的值, 这时候可能取到和target相等的位置.

但是当换成≤时, 当nums[mid] == target时, 仍然会进入if条件, left的值为mid+1后的值必然是大于target的值, 这时候就不可能取到和target相等的位置. 就会收敛到第一个大于target的位置. 

这点还是有点复杂的, 可以好好多看几遍.

第5条是说数组由递增改为递减, 其实只要将判断条件反过来就可以了, 还是很好理解的.

区间的变化完全取决于区间的定义, 因为左闭, 所以左区间为left = mid+1, 因为右开, 所以右区间变换为right = mid, 其实真实取的值也就是mid-1

最后的返回值上文已经说过了, 随便返回哪个都可以.

这个模板算是比较好记忆, 也比较简洁的模板了, 如果你也比较习惯左闭右开的区间, 直接理解记住此模板就可以了, 如果你习惯左闭右闭的区间, 那稍微更改下也就可以了.

class Solution: def searchInsert(self, nums: List[int], target: int) -> int: left, right = 0, len(nums) #采用左闭右开区间[left,right) while left < right: # 右开所以不能有=,区间不存在 mid = left + (right - left)//2 # 防止溢出, //表示整除 if nums[mid] < target: # 中点小于目标值,在右侧,可以得到相等位置 left = mid + 1 # 左闭,所以要+1 else: right = mid # 右开,真正右端点为mid-1 return left # 此算法结束时保证left = right,返回谁都一样