这是今天遇到的一个题,现在分享给大家。
用二分法,查找一个有序列表中的元素是否存在。有两种方法,一种是while循环,一种是递归。
代码如下:
方法一:
# -*- coding = UTF-8 -*-
def BinarySearch(l, n):
low = 0
high = len(l) - 1
while low <= high: #当要查找的数刚好在两端时,条件必须有low=high,不然无法查到两端的数
mid = (low + high) // 2
if l[mid] < n:
low = mid + 1 #当n在列表的右半边时,右边界的起始值是mid+1下标所对应的元素
elif l[mid] > n:
high = mid - 1 #当n在列表的左半边时,左边界的结束值为mid-1下标所对应的元素
else:
return mid, l[mid]
return None
# sorted array
L = [3, 4, 7, 11, 13, 17, 20, 108]
print(BinarySearch(L, 3))
print(BinarySearch(L, 99))
print(BinarySearch(L, 13))
方法二:
# -*- coding: utf-8 -*-
"""
Created on Sat Aug 13 15:32:09 2018
@author: h_pw
"""
def search(lst,num,lower=0,upper=None):
if upper is None: upper = len(lst)-1
middle = (lower + upper) // 2
if num == lst[middle]:
return middle,lst[middle]
elif num < lst[middle]:
return search(lst,num,lower,middle)
else:
return search(lst,num,middle+1,upper)
if __name__ == '__main__':
L = [2,5,6,12,17,18,29,30]
n = 2
result = search(L,n)
print(result)
ps:使用二分查找法的前提是列表必须是有序列表。