思路:由于该题给的数组是一个有序数组,要求寻找有序数组的目标值,符合二分查找的特点,因此本题使用二分查找
1.如果寻找到的目标值等于数组的中位数,直接返回该中位数的下标
2.如果目标值大于中位数,查找中位数右边的数组,将左边界指针l向中位数右移一位
3.如果目标值小于中位数,查找中位数左边的数组,将右边界指针r向中位数左移一位
注意:存在四种情况
第一种:目标值在数组中,直接返回目标值的下标
第二种:目标值不在数组中,目标值在两数之间,返回下标值l即为插入的位置
第三种:目标值不在数组中,目标值在最后一个元素之后,返回下标值l即为插入的位置
第四种:目标值不在数组中,目标值在第一个元素之前,返回下标值l即为插入的位置
class Solution:
def searchInsert(self, nums: List[int], target: int) -> int:
l = 0
r = len(nums) - 1
while l <= r:
mid = (l + r) // 2
if target == nums[mid]:
return mid
elif target > nums[mid]:
l = mid + 1
else:
r = mid -1
return l