我想查找排序数组中是否存在数字。直说,一个数组包含从1到63的斐波那契数。下面是斐波那契数生成器及其一些输出。
stacksize = 10000 # default 128 stack
from functools import lru_cache
@lru_cache(stacksize)
def nthfibonacci(n):
if n <= 1:
return 1
elif n == 2:
return 1
elif n > 2:
return nthfibonacci(n - 2) + nthfibonacci(n - 1)
output = [nthfibonacci(k) for k in range(1,63+1)]
# truncated output: [1, 1, 2, 3, 5, 8, 13, 21, 34, 55, 89, 144, 233, 377, 610,987,
1597, 2584, 4181, 6765, 10946, 17711, 28657, 46368,.....]
现在我想查找数字7 是否存在,所以我使用了以下代码 使用python bisection module
from bisect import bisect_left
elem_index = bisect_left(a=output, x=7, lo=0, hi=len(arr) - 1)
# output of elem_index is 5 ???? . But it is expected to be len(output) +1, right?
# as we know if element is not found it returns len(array) +1
如果我只是写一个简单的二进制搜索,它也会给我正确的结果,如下所示:
def binsearch(arr, key):
# arr.sort()
low = 0
high = len(arr) - 1
while low <= high:
mid = (low + high) // 2
if arr[mid] == key:
return mid
else:
if arr[mid] < key:
low = mid + 1
else:
high = mid - 1
return -1
print(binsearch(arr, 7)) # it gives me -1 as expected
那是怎么回事?
答案 0 :(得分:1)
jsfiddle解释了该行为:
bisect_left(...) bisect_left(a, x[, lo[, hi]]) -> index Return the index where to insert item x in list a, assuming a is sorted.
简而言之,bisect_left
(和bisect_right
)会告诉您元素是否存在,如果不存在则将其插入。
考虑一个人为的例子。当该值存在时,让我们在排序列表中搜索一个值。
l = [1, 4, 5]
bisect.bisect_left(l, 4)
# 1
bisect_left
返回1,因为l[1]
是4
。现在,重复该过程,但使用该列表中不存在的值。
bisect.bisect_left(l, 3)
# 1
在这种情况下,l[1]
是您在该排序列表中找到3个(如果存在)的地方。
这对您意味着什么?,您要做的就是修改函数以查询bisect_left
返回的索引处的元素,
def binary_search(items, key):
idx = bisect_left(items, key)
if items[idx] != key:
return -1
return idx