二分查找及相关扩展

# 查找值为k的元素
def findX(data,k):
    lo = 0
    hi = len(data)-1
    while lo<=k:
        # 这样获得的mid一定比较靠近lo
        mid = lo + (hi-lo)//2
        if data[mid] == k:
            return mid
        if data[mid]>k:
            hi = mid - 1
        else:
            lo = mid + 1
    return -1

# 找最后一个大于某个值、或第一个小于某个值的没有意义（一定是首或尾）

# 查找最后一个小于目标值的数
# 因可能答案为mid，所以返回hi (这种情况hi会等于lo-1)
# 举个例子data = 1 3 9 10 10 17  k = 10
def findLastL(data,k):
    lo = 0
    hi = len(data)-1
    while lo<=hi:
        mid = lo+(hi-lo)//2
        if data[mid]>=k:
            hi = mid-1
        else:
            lo = mid+1
    return hi if hi>=0 else -1

# 查找最后一个小于等于目标值
def findLastLeq(data, k):
    lo = 0
    hi = len(data) - 1
    while lo <= hi:
        mid = lo + (hi - lo) // 2
        if data[mid] > k:
            hi = mid - 1
        else:
            lo = mid + 1
    return hi if hi >= 0 else -1

# 查找第一个大于目标值的数
def findFirstG(data,k):
    lo = 0
    hi = len(data)-1
    while lo<=hi:
        mid = lo+(hi-lo)//2
        if data[mid]<= k:
            lo = mid+1
        else:
            hi = mid-1
    return lo if lo<=len(data)-1 else -1

# 查找第一个大于等于目标的数
def findFirstG(data,k):
    lo = 0
    hi = len(data)-1
    while lo<=hi:
        mid = lo+(hi-lo)//2
        if data[mid] < k:
            lo = mid+1
        else:
            hi = mid-1
    return lo if lo<=len(data)-1 else -1