刚开始看《算法图解》,上机操作了下二分查找,结果马上就出问题了……
代码如下
def binary_search(list, item):
low = 0
high = len(list)-1
while low <= high:
mid = (low + high) // 2
guess = list[mid]
if guess == item:
return mid
if guess > item:
high = mid - 1
else:
low = mid + 1
return None
my_list = [1, 3, 5, 7, 9]
print (binary_search(my_list, 1))
咋一看好像没问题,但运行结果几乎全是None,除了查找的数是5的情况。
再仔细的观察代码(用的在线开发环境没法调试),总算是发现return None
的位置好像不太对。
调整成下面这样就正确了:
def binary_search(list, item):
low = 0
high = len(list)-1
while low <= high:
mid = (low + high) // 2
guess = list[mid]
if guess == item:
return mid
if guess > item:
high = mid - 1
else:
low = mid + 1
return None
my_list = [1, 3, 5, 7, 9]
print (binary_search(my_list, 1))
错误分析
原来的代码错误地把return None
和while low <= high:
对齐了,这导致return None
被包含进了while循环里,所以当查找的数为第一个list[mid]的值也就是5的时候,此时正好guess == item
因此才会返回正确的结果,而其他的情况都会在第一次循环的时候就把return None
给执行了导致结果几乎全是None。
反思
由于Python代码的特性,今后要特别留意Python代码的对齐问题。