如果找不到值，则在二进制搜索中返回较小的数字

Question

我有一个排序后的数字列表。 [1,2,4,6,10,12]

我想在列表中找到一个数字。如果找不到，我想返回下一个最小的数字。

例如，n = 8。我搜索 8 但找不到它。然后我将返回 6，因为它是下一个最小的数字。

我有一些代码，但似乎无法正确索引：

def search_val(data, n):
    low = 0
    high = len(data) - 1
    if data[-1] <= n:
        return data[-1]
    if data[0] >= time:
        return

    while low < high:
        mid = (low + high) / 2
        if data[mid] == n:
            return data[n]
        if data[mid] > n:
            high = mid -1
        if data[mid] < n:
            low = mid + 1
    if data[low] < n:
        return data[low]
    else:
        return data[low - 1]

Answer 1

这将解决您的所有问题，并且应该更快：

def search_val(data, n):
    low = 0
    high = len(data) - 1

    while low <= high:
        mid = (low + high) // 2

        if data[mid] > n:
            high = mid -1
        elif data[mid] < n:
            low = mid + 1
        else:
            return data[mid]

    # Now low > high

    if high >= 0:
        return data[high]

    # All values are > n.  Just return the first.  Could
    # be changed to return None.

    return data[0]

注意，在所有值都是> n 的情况下，它返回第一个值。如果需要，您当然可以将其更改为返回 None。

还要注意它假定len(data) 是>= 1。如果情况并非总是如此，您可以在顶部添加一个检查，如果 data 是空列表，则返回 None。

Answer 2

列表是否保证有序？如果没有，试试这个：

data = [1,2,4,6,10,12]
result = max((n for n in data if n <= 8))