使用二分查找法,返回值的索引
方法一:
# 返回 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 = [1, 3, 4, 10, 15, 20]
x = 20
# 函数调用
result = binarySearch(arr, 0, len(arr) - 1, x)
if result != -1:
print("元素在数组中的索引为 %d" % result)
else:
print("元素不在数组中")
方法二:
"""二分查找"""
n = 10
arr = [1, 3, 4, 10, 15, 20, 33]
mid_index = int(len(arr) / 2)
index = mid_index
mid = arr[mid_index]
while True:
if n == mid:
print('元素在数组中的索引{}'.format(index))
break
elif index == 0 or index == (len(arr) - 1):
print('元素在数组中不存在')
break
elif index > mid_index and n < mid:
index = index - int((len(arr) - index) / 2)
mid = arr[index]
elif index > mid_index and n > mid:
index = index + int((len(arr) - index) / 2)
mid = arr[index]
elif index <= mid_index and n < mid:
index = int(index / 2)
mid = arr[index]
else:
index = index + int(index / 2)
mid = arr[index]