35. 搜索插入位置

问题链接

https://leetcode.cn/problems/search-insert-position/description/

解题思路

搜索插入位置，是一个常见的二分算法。

二分是有固定模板的。这个题目是搜索插入位置，和最简单的二分不同。

不同之处在于，它在查找失败时不返回False，而是返回插入位置，这就要求我们能够优雅的查找到插入位置。

首先我们按照二分的一般思路写好代码。然后我们去想，如果此时left == right ，即这是最后一轮比较的话，如果target < mid，那代表target应该插入mid当前的位置上，此时指向mid的是left和mid。

如果target > mid , 此时target应该插入在mid后，此时指向mid+1的只有left。所以我们直接返回left即可。

代码

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