該方法當找不到元素的時候,不是簡單的返回-1,而是同時返回鄰近的下標。該下標代表的數值可能大於,也可能小於指定的元素。
#coding:utf8
def bSearch(array,element):
low = 0
high = len(array) - 1
while low <= high:
k = (low + high) >> 1
if array[k] < element:
low = k + 1
elif array[k] > element:
high = k - 1
else:
return k,-1 #有匹配返回下標和-1
return -1,k #沒有匹配返回-1 和鄰近的下標,可能大於,也可能小於
array = [1, 3, 5, 7, 9, 6, 8, 0]
print(bSearch(array, 5))