【leetcode100】搜索插入位置

1、题目描述

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

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

示例 1:

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

示例 2:

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

2、初始思路

2.1 思路

使用二分查找法，左闭右闭

2.2 代码

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