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)-1def search(l,r,target):mid = (l+r) // 2if l>r:return lif nums[mid] == target:return midelif 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)-1while (l<=r):mid = (l + r) // 2if nums[mid] > target:r = mid -1elif nums[mid] < target:l = mid + 1else:return midreturn l