题目:
"""
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。
如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
"""
法一:比较法
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)