前提:升序數組,待查元素在數組中。
二分查找:就是一個遞歸函數c。待查元素a,當前數組中位數b,如果b=a則返回b的索引,b>a則在b左側的子數組中調用函數c,否則在b右側子數組中調用函數c。
第一次思考,按著上面的思路編程后的結果:
1
2
3
4
5
6
7
|
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) |
第二次思考,簡化中位數計算邏輯:
1
2
3
4
5
6
7
|
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)
以上就是二分查找變為“一行代碼”版的過程。
運行測試:
1
2
3
|
if __name__ == '__main__': a = [1, 2, 33, 43, 52, 66, 88, 99, 111, 120] print(f"Target index: {binary_search(0, a, value=33)}") |
結果如下:
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持服務器之家。
原文鏈接:https://www.cnblogs.com/LanTianYou/p/8907354.html