1. 程式人生 > >Python 二分查找

Python 二分查找

temp light bis 返回值 arch from sort 應該 位置

Python 中自帶了bisect模塊實現了二分查找

def merge_search(list, num):
    list.sort()
    start, end = 0, len(list)-1  # 數組下標
    while start < end:
        mid = (start + end) // 2
        if list[mid] < num:
            start = mid + 1      # 更新數組下標
        elif list[mid] > num:
            end = mid - 1        # 更新數組下標
        else:
            return mid
    return -1


temp = [1, 2, 0, 5, 7]
num = [5, 4]
for i in num:
    print(merge_search(temp, i))
    from bisect import bisect_left, bisect_right, bisect
    print(bisect_left(temp, i))    # 在列表中查詢目標值num, num存在時返回X左側的位置, 不存在則返回X應該插入的位置, 返回值為index
    print(bisect_right(temp, i))   # 在列表中查詢目標值num, num存在時返回X右側的位置, 不存在則返回X應該插入的位置, 返回值為index
    print(bisect(temp, i))    # <=> bisect_right

3
3
4
4
-1
3
3
3

  

Python 二分查找