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