给定一个 n
个元素有序的(升序)整型数组 nums
和一个目标值 target
,写一个函数搜索 nums
中的 target
,如果目标值存在返回下标,否则返回 -1
。
示例:
输入:nums = [-1,0,3,5,9,12],target = 9
输出:4
输入:nums = [-1,0,3,5,9,12],target = 2
输出:-1
完整的思路如《代码随想录》所示
解法一:左闭右闭区间
class Solution:
def search(self, nums: List[int], target: int) -> int:
left, right = 0, len(nums) - 1 # 定义target在左闭右闭的区间里,[left, right]
while left <= right:
middle = left + (right - left) // 2
if nums[middle] > target:
right = middle - 1 # target在左区间,所以[left, middle - 1]
elif nums[middle] < target:
left = middle + 1 # target在右区间,所以[middle + 1, right]
else:
return middle # 数组中找到目标值,直接返回下标
return -1 # 未找到目标值
接下来,我将用自己的理解解释以上代码,过于基础的我就不做赘述了
left, right = 0, len(nums) - 1 # 定义target在左闭右闭的区间里,[left, right]
将left和right 分别放在数组的左右两个边界上,它和解法二的区别在于,我们是左闭右闭区间,因此左右两个边界都要能取到具体的值,所以这里的right用len(nums)-1表示。如果数组总共有7个数字,那len(nums)-1就等于6,由于右侧为闭区间,所以我们能实际取到下标为6的数,即数组中的第7个数。
while left <= right:
middle = left + (right - left) // 2
这里的意思是,只要在没找到target的下标前,只要left小于等于right就可以一直循环。每次循环中都要重新给left和right之间的中间数middle赋值,middle为left的值加上right和left之间的差值除以2后的整数部分。
后续代码注释的已经很清楚了,如果target小于数组中下标为middle的值,那就代表target在数组二分后的左半部分,那就把左半部分当做新的数组继续进行二分(即右边界right=middle-1);如果target大于数组中下标为middle的值,那么同理把右半部分作为新的数组进行继续二分。不停进行二分直到最后找到下标为target的值,未找到的话代表数组中没有值等于target,就return -1。
解法二:左闭右开区间
class Solution:
def search(self, nums: List[int], target: int) -> int:
left, right = 0, len(nums) # 定义target在左闭右开的区间里,即:[left, right)
while left < right: # 因为left == right的时候,在[left, right)是无效的空间,所以使用 <
middle = left + (right - left) // 2
if nums[middle] > target:
right = middle # target 在左区间,在[left, middle)中
elif nums[middle] < target:
left = middle + 1 # target 在右区间,在[middle + 1, right)中
else:
return middle # 数组中找到目标值,直接返回下标
return -1 # 未找到目标值
与解法一不同的点在于,解法二把区间设置为左闭右开,即区间右边界实际上是取不到的。可以看到,代码中用len(nums)去赋值right,数组如果有7个数的话,那right就等于7,但下标为7的数是不存在的,即类似区间[0,7),你只能取到6。
因此在后续的代码中也有相应变化,存在于target在二分后的左半区间内,因为区间左闭右开,middle反正也取不到(不同于解法一左闭右闭区间,下标为middle的值实际上能取到,因此将middle-1赋值给right的情况)。解法二中middle反正也取不到,实际上只能从middle-1开始取,因此将middle赋值给right就可以了。