python leetcode35:搜索插入位置

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

法一：比较法

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        for i in range(len(nums)):
            if nums[i] == target: #相等
                return i
            elif i < len(nums)-1 and target > nums[i] and target < nums[i+1] : #target处于中间位置
                return i+1
            elif i == 0 and target < nums[i]: #开头
                return i
        return len(nums) #不存在于数组

法二：二分查找

直接套用二分法即可，即不断用二分法逼近查找第一个大于等于 \textit{target}target 的下标 。

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


nums = [1, 3, 5, 6]
target = 7
S = Solution()
result = S.searchInsert(nums,target)
print(result)