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