繁体   English   中英

在排序列表中查找大于给定数字的最小数字

[英]Find the smallest number that is greater than a given number in a sorted list

给定一个排序的数字列表,我需要找到大于给定数字的最小数字。 考虑这个清单:


arr=[1,2,3,5,7,11,101,131,151,181,191,313,353,373,383]

假设指定的数字是320.然后,我的方法应该返回353,因为353是大于320的最小数字。

我试图使用略微修改的二进制搜索形式; 但是在执行时,程序进入无限循环。


def modBinarySearch(arr,x):
    l=len(arr)
    mid=l/2
    if arr[mid]>=x and arr[mid-1]<x:
        return arr[mid]
    elif arr[mid]>x and arr[mid-1]>x:
        modBinarySearch(arr[mid:l],x)
    else: 
        modBinarySearch(arr[0:mid],x)

N=int(raw_input())
arr=[1,2,3,5,7,11,101,131,151,181,191,313,353,373,383]
print modBinarySearch(arr,N)

有人可以指出我做错了什么吗?

有一个标准模块, bisect ,已经这样做:

In [49]: arr[bisect.bisect(arr, 320)]
Out[49]: 353

我认为这应该是搜索排序列表的首选方法。 手册中有几个例子

至于你的实现,有很多问题:

  1. 您的递归不能正确处理小数组。
  2. 在第二个分支中完成的切片是不正确的。
  3. 你的功能不会返回任何东西。
  4. 因为arr是按升序排列的, arr[mid]>x and arr[mid-1]>x相当于arr[mid-1]>x ,这表明你没有写出你的意思

最后但并非最不重要的是,递归和所有切片对于这个问题完全没有必要。

如果列表的大小为15,则完全抛弃二进制搜索并使用顺序搜索。

您会发现代码更容易编写,除非您需要每秒执行数百万次,否则顺序解决方案将足够快。

如果确实需要坚持使用二进制搜索,你的第一步应该是实际回报您的递归调用的结果。

如果arr [mid]和arr [mid-1]两者都大于你的数字,你应该搜索arr [0:mid],你不觉得吗?

 elif arr[mid]>x and arr[mid-1]>x:
    modBinarySearch(arr[0:mid],x)
else: 
    modBinarySearch(arr[mid:1],x)
def modBinarySearch(arr, n):
    m = len(arr) / 2

    if arr[m] >= n and arr[m - 1] < n:
        return arr[m]
    elif arr[m] > n and arr[m - 1] > n:
        return modBinarySearch(arr[:m], n)
    else:
        return modBinarySearch(arr[m:], n)


arr = [1, 2, 3, 5, 7, 11, 101, 131, 151, 181, 191, 313, 353, 373, 383]
n = 320
print(modBinarySearch(arr, n))
     python 3.2

     next(i for  i in arr if i>320)

bisect模块是您的最佳选择:

from bisect import bisect_left, bisect_right

arr=[1,2,3,5,7,11,101,131,151,181,191,313,353,373,383]


def find_lt(a, x):
    'Find rightmost value less than x'
    i = bisect_left(a, x)
    if i:
        return a[i-1]
    raise ValueError

def find_gt(a, x):
    'Find leftmost value greater than x'
    i = bisect_right(a, x)
    if i != len(a):
        return a[i]
    raise ValueError

print find_lt(arr,320)          
print find_gt(arr,320)  

版画

313
353     

如果列表已排序:

x = range(20)
N= 15

for i in x:
    if i>N:
        print i
        break

给16。

如果使用numpy:

x = np.arange(20)
N = 15
x[x>15][0]

给16。

如果您正在寻找简单的实现,针对您的具体问题,让我重温一下。

def modBinarySearch(arr,x):
    l=len(arr)
    mid=l/2
    if arr[mid] >= x and arr[mid-1] < x:
        return arr[mid]
    elif arr[mid]>x and arr[mid-1]>x:
        num = modBinarySearch(arr[0:mid],x)
    else:
        num = modBinarySearch(arr[mid:l],x)
    return num

N=int(raw_input('Enter a number: '))
arr=[1, 2, 3, 5, 7, 11, 101, 131, 151, 181, 191, 313, 353, 373, 383]
print modBinarySearch(arr,N)

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM