python二分查找的相关代码
二分查找
二分查找是非常基本的算法题,一定要能熟练写出
def myBinarySearch(arr,target):
##边界检查
if len(arr) < 1:
return -1
##注意数组的左右边界
lo = 0
high = len(arr) - 1
##为什么市lo <= high是因为我们要比较到数组只有一个元素,这个时候lo == high,此时 mid == lo == high,这个时候arr[mid] != target,说明查找失败
while lo <= high:
mid = (lo + high) // 2
if arr[mid] == target:
return mid
elif arr[mid] < target:
lo = mid + 1
else:
high = mid - 1
return -1
找出大于等于目标值的最小索引
def my_lower_bound(arr,target):
##边界检查
if len(arr) < 1:
return -1
lo = 0
high = len(arr) - 1
##注意这里为什么是 <,因为最后循环结束的时候只能lo == high,并且mid == high == lo,?arr[mid] >= target
while lo < high:
mid = (lo + high) // 2
## target 比中间之要小,说明最小只在[mid+1,high]
if arr[mid] < target:
lo = mid + 1
##taget比中间值要大,考虑到重复值,high不能是mid - 1
else:
high = mid
return lo
返回大于目标值的最小索引(第一个大于目标值的索引)
def my_upper_bound(arr,target):
if len(arr) < 1:
return -1
lo = 0
high = len(arr) - 1
while lo < high:
mid = (lo + high) // 2
if arr[mid] <= target:
lo = mid + 1
else:
high = mid
return lo