优质博客:详解二分查找算法
看完了没有搞不定的二分查找
二分查找的三种写法:
写法一:传入right为数组长度
class Solution:
def search(self, nums: List[int], target: int) -> int:
left = 0
right = len(nums)
while(left < right): # 注意点1
mid = int((left + right) / 2)
if(nums[mid] == target):
return mid
elif(nums[mid] < target):
left = mid + 1 # 注意点2
elif(nums[mid] > target):
right = mid # 注意点3
return -1
注意点1: 因为传入搜索范围为[left, right),最右边的数不需要搜索,所以这里是小于;
注意点2: 因为mid判断后,需要拆分为两个[left, right)区间,给left赋值要是需要判断的,而mid不用再判断,故给left赋值需要+1;
注意点3: 因为mid判断后,需要拆分为两个[left, right)区间,给right赋值是不需要判断的,而mid不用再判断,所以给right赋值时不需要-1,直接给mid值就可以。
标题写法二:传入right为数组长度-1
class Solution:
def search(self, nums: List[int], target: int) -> int:
left = 0
right = len(nums) - 1
while(left <= right): # 注意点4
mid = int((left + right)/2)
if(nums[mid] == target):
return mid
elif(nums[mid] < target):
left = mid + 1 # 注意点5
elif(nums[mid] > target):
right = mid - 1 # 注意点6
return -1
注意点4: 因为传入搜索范围为[left, right],最右边的数需要搜索,所以这里是小于等于;
注意点5: 因为mid判断后,需要拆分为两个[left, right]区间,给left赋值要是需要判断的,而mid不用再判断,故给left赋值需要+1;
注意点6: 因为mid判断后,需要拆分为两个[left, right]区间,给right赋值是需要判断的,而mid不用再判断,所以给right赋值时需要-1。
写法三:递归调用
search是调用入口:
class Solution:
def binary_search(self, nums, left, right, target):
if(left < right):
mid = int((left + right) / 2)
if(nums[mid] == target):
return mid
elif(nums[mid] < target):
left = mid + 1
return self.binary_search(nums, left, right, target)
elif(nums[mid] > target):
right = mid
return self.binary_search(nums, left, right, target)
return -1
def search(self, nums: List[int], target: int) -> int:
left = 0
right = len(nums)
return self.binary_search(nums, left, right, target)
这里递归调用binary_search的传入参数可以参照写法一和写法二有两种写法。