题目:给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
我们在排序数组中寻找target就可以用二分查找算法,因此代码就可以套用上个章节的内容。这块和上章节不同的地方上章节提到如果目标值不存在返回-1,而这道题目标值不存在数组中返回它将会被按顺序插入的位置,因此代码需要做细微的调整当查找不到相关内容的时候需要返回mid+1
class soultion:
#二分查找算法函数
def binary(self, s,target):
#定义左右指针
left=0
right = len(s)-1
mid=int(left+(right-left)/2)
while(left<len(s) and right>=0 and right>=left):
#二分查找算法计算
if(s[mid]<target):
left=mid+1
elif(s[mid]>target):
right=mid-1
elif(s[mid]==target):
return mid
#mid值的定义
mid=int(left+(right-left)/2)
#需要替换-1到mid+1
return mid+1
soultion = soultion()
s = [-1,0,3,5,9,12]
target=12
m = soultion.binary(s,target)
print(m)