Leetcode704-二分查找
思路:二分法。
定义left,right,初始化为第一个和最后一个元素。
mid=(left+right)// 2,如果nums[mid]>target,中间的值比目标值大,说明目标值只可能出现在当前中间位置的左方,所以right=mid-1.
最开始写的时候犯了一个错误:写成了right=mid,这种情况下,会陷入死循环。
比如([-1,0,5],-1)
查找-1时:
left = 0
right = 2
mid = (0+2)//2=1
nums[mid] = nums[1]=0, 0>-1, 所以right=mid=1(这是错误的),按照这样,新的mid=(0+1)//2 = 1,nums[mid] = nums[1]=0, 0>-1,死循环。
正确的应该是right=mid-1
比如([-1,0,5],-1)
查找-1时:
left = 0
right = 2
mid = (0+2)//2=1
nums[mid] = nums[1]=0, 0>-1, 所以right=mid-1=0
left = 0,right=0,所以mid=0,此时nums[mid]正好等于target,返回mid。
另外一个需要注意的点就是while条件,需要写成while left<=right,最开始写成了while left<right,在上面这种left = 0,right=0条件下,会直接返回-1,得不到正确结果。
代码:
class Solution:
def search(self, nums: List[int], target: int) -> int:
left = 0
right = len(nums) - 1
while left <= right:
mid = (left + right) // 2
if target > nums[mid]:
left = mid + 1
elif target < nums[mid]:
right = mid - 1
else:
return mid
return -1