Python 二分查找
二分搜索是一種在有序數組中查找某一特定元素的搜索演算法。搜索過程從數組的中間元素開始,如果中間元素正好是要查找的元素,則搜索過程結束;如果某一特定元素大於或者小於中間元素,則在數組大於或小於中間元素的那一半中查找,而且跟開始一樣從中間元素開始比較。如果在某一步驟數組為空,則代表找不到。這種搜索演算法每一次比較都使搜索範圍縮小一半。
實例 : 遞歸
# 返回 x 在 arr 中的索引,如果不存在返回 -1
def binarySearch (arr, l, r, x):
# 基本判斷
if r >= l:
mid = int(l + (r - l)/2)
# 元素整好的中間位置
if arr[mid] == x:
return mid
# 元素小於中間位置的元素,只需要再比較左邊的元素
elif arr[mid] > x:
return binarySearch(arr, l, mid-1, x)
# 元素大於中間位置的元素,只需要再比較右邊的元素
else:
return binarySearch(arr, mid+1, r, x)
else:
# 不存在
return -1
# 測試數組
arr = [ 2, 3, 4, 10, 40 ]
x = 10
# 函數調用
result = binarySearch(arr, 0, len(arr)-1, x)
if result != -1:
print ("元素在數組中的索引為 %d" % result )
else:
print ("元素不在數組中")
執行以上代碼輸出結果為:
元素在數組中的索引為 3