python: 数列与二分法（LeetCode题目）

题目来自LeetCode：

二分法是高中数学的知识，但在使用二分法解决问题的时候我们需要注意：数列必须是排好序、且数列中的数字无重复，否则会返回多个下标。

1. 35

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。

请必须使用时间复杂度为 O(log n) 的算法。

示例 1:

输入: nums = [1,3,5,6], target = 5
输出: 2
示例 2:

输入: nums = [1,3,5,6], target = 2
输出: 1
示例 3:

输入: nums = [1,3,5,6], target = 7
输出: 4

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/search-insert-position
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

代码如下：

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        
        left = 0
        right = len(nums) - 1
        while left <= right:
            mid = (left + right) // 2
            if nums[mid] > target:
                right = mid - 1
            elif nums[mid] < target:
                left = mid + 1
            else:
                return mid
        return right + 1

以上代码段属python3。

二分法的逻辑很容易思考，但写出来的代码非常容易错，主要原因是对于边界的定义不够清楚。

二分法有两种：1. 左右全包，即：[left, right]; 2. 包左不包右，即：[left, right)

两个方法思路差不多，但代码有一点区别。我倾向于用第一种方法。

思路：

假设在该数组中有两个指针，分别指向最左端（left）和最右端（right），其对应的值为列表索引的最大值和最小值，即：left = 0；right = len(nums) - 1，这两个属于循环不变量，在循环中永远指向两端。（但不代表其值不变）

二分法需要添加一个中间量mid，不过这个中间量随循环改变，因此放在循环内。

其值为：mid = left + (right - left) // 2    # 即 mid = (right + left) // 2

// 在python中如果有余数，直接切掉，比如下面举例的数列中，mid会指向5， 而不是6.

以下是mid的作用：( | 代表mid）假设 target = 7。

 0  1  2  3  4  5

[1, 3, 5, 6, 8, 9]

 L       |          R

（第一次循环时，比较： nums[mid] 和 target谁大谁小，结果显而易见，此时，L 就可以移动至 mid + 1 的位置了。）

[1, 3, 5, 6, 8, 9]

              L  |  R

（第二次循环，发现 nums[mid] > target,  同理，R向左移动，R = mid - 1）

于是会出现：

[1, 3, 5, 6, 8, 9]

              L  

              R

此时数列已经循环结束，发现数列内没有target，于是我们要按照顺序将target放到数列内，并返回其索引，到这里，target的索引显而易见：right + 1 或者是 left + 1。