题目描述
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
【题目来源:力扣(LeetCode)】
二分查找
百科:二分查找也称折半查找(Binary Search),它是一种效率较高的查找方法。首先,假设表中元素是按从小到大排列,将表中间mid位置记录的关键字与查找关键字比较,如果两者相等,则查找成功;否则利用中间位置记录将表分成前、后两个子表,如果中间位置记录的关键字大于查找关键字,则进一步查找前一子表,否则进一步查找后一子表。重复以上过程,直到找到满足条件的记录,使查找成功,或直到子表不存在为止,此时查找不成功。
对于该题
定义出l,r作为列表nums的首位和尾位,取mid为(l+r)//2作为nums的中间位置。将nums[mid]与target进行比较,因为nums是升序排列,所以我们可以根据nums[mid]和target的大小决定l或r的移动方向以做下一步的对比,移动时mid不断改变,直到nums[mid]等于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 nums[mid] == target:
return mid
elif nums[mid] > target:
r = mid - 1
else:
l = mid + 1
return l