我一直在尝试递归地编写二进制搜索。当我使用list [:]语法执行此操作时,由于出现几个错误或没有获得正确的值,因此无法获得预期的结果。
def binary_search(arr, val):
left = 0
right = len(arr)-1
mid = (left + right)//2
#Go through the array
if (val == arr[mid]):
return mid
#Check right side of arr
if (val > arr[mid]):
return binary_search(arr[mid+1:], val)
#Check left side of arr
return binary_search(arr[:mid], val)
编辑:示例输入和输出
arr1 =[]
for i in range(10):
arr1.append(i)
for i in range(10):
print(binary_search(arr1,i))
我希望得到类似'0,1,2,3,4,5,6,7,8,9'
的东西,但得到'0,1,0,0,4,None ,None,2,0,0'
答案 0 :(得分:2)
您有两个问题。第一个是错字,你说
if (val > mid):
你应该说
if (val > arr[mid]):
因为您要比较的是值而不是索引。
第二个更细微...当您检查数组的右侧时,位于:
return binary_search(arr[mid+1:], val)
您要传递给递归调用(arr[mid+1:]
)的子数组已经在数组中间开始,这意味着递归调用的结果将返回子数组中元素的索引。因此,您需要添加用于拆分数组的索引增量,以再次基于完整数组创建索引:
return binary_search(arr[mid+1:], val) + (mid + 1)
以下是完整的完整代码:
def binary_search(arr, val):
left = 0
right = len(arr)-1
mid = (left + right)//2
#Go through the array
if (val == arr[mid]):
return mid
#Check right side of arr
if (val > arr[mid]):
return binary_search(arr[mid+1:], val) + (mid + 1)
#Check left side of arr
return binary_search(arr[:mid], val)
答案 1 :(得分:0)
您正在将val
与mid
(而不是arr[mid]
)进行比较。另外,如果您使 ifs 互斥,它会更好看。另外,按照下面的nosklo的回答,您需要为大于情况添加索引偏移量:
#Go through the array
if (val == arr[mid]):
return mid
#Check right side of arr
elif (val > mid):
return binary_search(arr[mid+1:], val) + (mid + 1)
#Check left side of arr
else:
return binary_search(arr[:mid], val)