二分查找关键点在于 查找范围
,只要查找范围能找对,不至于漏掉,思路就对了一半,剩下的一半则在于 不要死循环
,搞清楚这两点基本没问题,我常用的是左闭右闭
模板如下:
l = # 左边界
r = # 右边界
# 按照左闭右闭查找
while l <= r:
mid = (l + r) // 2
if f(mid) < condition:
l = mid + 1
elif f(mid) > condition:
r = mid - 1
else:
# 条件满足时触发。。
return mid
这个模板的好处是基本不会出错,且通用,难点在于需要清楚具体return 什么内容(一般以r
为主)。
· 查找左/右边界(端点)
# 找左边界
l = 0
r = len(nums) - 1
l_ans = -1
while l <= r:
mid = (l + r) // 2
if nums[mid] < target:
l = mid + 1
elif nums[mid] > target:
r = mid - 1
else:
# 找左边界
l_ans = mid
r = mid - 1
if l_ans < 0:
return r + 1
else:
return l_ans
# 找右边界
l = 0
r = len(nums) - 1
l_ans = -1
while l <= r:
mid = (l + r) // 2
if nums[mid] < target:
l = mid + 1
elif nums[mid] > target:
r = mid - 1
else:
# 找右边界
r_ans = mid
l = mid + 1
if l_ans < 0:
return r + 1
else:
return l_ans
查找左右边界也是一个难点,见Leetcode 34、35题,只需要注意条件触发时的处理即可。
他山之玉:
优秀的二分查找总结:来自Github
二分查找优秀题解及总结:来自Leetcode-liweiwei1419