【算法合集-leetcode系列】3. 搜索插入位置（No.35）

题目描述： 

35. 搜索插入位置https://leetcode.cn/problems/search-insert-position/用普通循环方法找肯定是可以的，但题目要求复杂度O(log n)，就意味着要使用二分法啦。

整体思路类似于二分查找目标target，但本问题多了“插入”的情况，需注意边界条件问题。

我的思路：

（1）先判断左右两端临界问题。若超出数组范围，那么直接比较最大最小值，返回位置

（2）当target存在于list中，直接返回

（3）当target不存在鱼list中并位于中间两数之间时，解为（L，R】，也就是说返回值在右边界处。（小于R，插入位置为R；等于R，插入位置也为R；且条件L一定<R）

代码：

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        if target<=nums[0]:#头
            return 0
        if target>nums[-1]:#尾
            return len(nums)    
        left,right=0,len(nums)-1
        #边界条件这里特别容易出错，一定注意。
        #需考虑两个特殊例子就是[1 3 5 6]和[1 3]，target=2
        while left<right:
            mid=left+(right-left)//2
            #print(mid)
            if nums[mid]>target:  
                right=mid
            if nums[mid]<target:
                left=mid+1
            if nums[mid]==target:
                #print('ouput') 
                return mid
        return right

其他解答，两种边界判断思路。这个解答写的很清楚，感觉思路比我的解答清晰，搬运如下，稍微加点个人理解：

【思路】
1.二分法就是折半查找，先选中间的数，如果目标值比中间值大，去右边找；如果目标值比中间值小，去左边找；直到找到目标或超出了边界条件为止。
2.二分法有两种写法，假设n为数组长度，一种是在[0, n - 1]这个左闭右闭区间内找，一种是在[0, n)这个左闭右开区间内找，边界条件稍有不同，选择哪种纯看个人喜好。
3.找中间值也有两种写法，一是mid =(left + right)// 2，二是mid = left + (right - left) // 2，后者可以避免直接相加导致的数值溢出，对于本题来讲不会溢出，选哪种都一样。
 

（1）第一种思路：左闭右闭，[0,n-1]
左指针 left = 0，右指针 right = n - 1
中间值 mid = (left + right) // 2，循环条件 left <= right【一定可以等于，因为最终解一定不是left，所以最后返回的时候+1，为更新后的left】
当 nums[mid] = target 时，找到目标，返回mid
当 nums[mid] > target 时，说明当前值偏大，右指针左移至 mid - 1
当 nums[mid] < target 时，说明当前值偏小，左指针右移至 mid + 1
跳出循环时，left 指向 target 插入位置，返回 left

解答代码：

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:
                return mid
            elif nums[mid] > target:
                right = mid - 1
            else:
                left = mid + 1
        return left

此思路下我的代码：

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        if target<=nums[0]:#头
            return 0
        if target>nums[-1]:#尾
            return len(nums)    
        left,right=0,len(nums)-1
        #边界条件这里特别容易出错，一定注意。
        #需考虑两个特殊例子就是[1 3 5 6]和[1 3]，target=2
        while left<=right:
            mid=left+(right-left)//2
            #print(mid)
            if nums[mid]>target:  
                right=mid-1
            if nums[mid]<target:
                left=mid+1
            if nums[mid]==target:
                #print('ouput') 
                return mid
        return left

（2）第二种思路： 左闭右开，【0,n）

左指针 left = 0，右指针 right = n
中间值 mid = left + (right - left) // 2，循环条件 left < right(右边界取不到)
当 nums[mid] = target 时，找到目标，返回mid
当 nums[mid] > target 时，说明当前值偏大，右指针左移至 mid
当 nums[mid] < target 时，说明当前值偏小，左指针右移至 mid + 1
跳出循环时，left 指向 target 插入位置，返回 left

解答代码：

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

感谢参考：

https://leetcode.cn/problems/search-insert-position/solution/35-sou-suo-cha-ru-wei-zhi-er-fen-fa-de-l-6dz5/