【二分查找】总结

二分查找关键点在于 查找范围,只要查找范围能找对,不至于漏掉,思路就对了一半,剩下的一半则在于 不要死循环,搞清楚这两点基本没问题,我常用的是左闭右闭模板如下:

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

猜你喜欢

转载自blog.csdn.net/LoveJSH/article/details/128552612