1. 程式人生 > >Pythonic版二分查找

Pythonic版二分查找

col __name__ elif bin 子數組 info lse pytho 元素

前提:升序數組,待查元素在數組中。

二分查找:就是一個遞歸函數c。待查元素a,當前數組中位數b,如果b=a則返回b的索引,b>a則在b左側的子數組中調用函數c,否則在b右側子數組中調用函數c。

第一次思考,按著上面的思路編程後的結果:

def binary_search(index, a, value):
    if a[(len(a) - 1) // 2] == value:
        return index + (len(a) - 1) // 2
    elif a[(len(a) - 1) // 2] < value:
        return binary_search(index + (len(a) - 1) // 2 + 1, a[(len(a) - 1) // 2 + 1:], value)
    
else: return binary_search(index, a[0:(len(a) - 1) // 2 + 1], value)

第二次思考,簡化中位數計算邏輯:

def binary_search(index, a, value):
    if a[len(a) // 2] == value:
        return index + len(a) // 2
    elif a[len(a) // 2] < value:
        return binary_search(index + len(a) // 2, a[len(a) // 2:], value)
    
else: return binary_search(index, a[0:len(a) // 2], value)

第三次思考,去掉return,改為lambda形式:

binary_search = lambda index,a,value: index + len(a) // 2 if a[len(a) // 2] == value else binary_search(index + len(a) // 2, a[len(a) // 2:], value) if a[len(a) // 2] < value else binary_search(index, a[0:len(a) // 2], value)

以上就是二分查找變為“一行代碼”版的過程。

運行測試:

if __name__ == __main__:
    a = [1, 2, 33, 43, 52, 66, 88, 99, 111, 120]
    print(f"Target index: {binary_search(0, a, value=33)}")

結果如下:
技術分享圖片

Pythonic版二分查找