python bisect-原创手记-慕课网

Python的bisect模块，用于维护有序列表。bisect模块实现了将元素插入一个有序序列的算法，它利用二分算法实现排序，相比于每次调用sort而言，这样无疑要高效许多。bisect_*系列方法用于求索引，insort_*系列方法用于有序插入

1. 常用方法

I. bisect.bisect_left(a, x, lo=0, hi=len(a))

a是一个有序列表，x是插入其中的元素， lo和hi分别代表列表的切片索引（默认为0和列表长度）

此方法用于寻找一个保证插入元素后列表仍然有序的索引值。当元素x插入a列表后列表仍保持有序状态，且当列表内已经有与x值相同的元素时，x会插入到其中所有相同元素的最左边，方法会返回所插入元素在列表中的索引，并保证在索引值左边的元素都比所插入元素小，右边的都不比所插入元素大(大于等于)

import bisect

if __name__ == '__main__':
    l = [1, 2, 3, 4]
    k = []
    print(bisect.bisect_left(l, 3))
    print(bisect.bisect_left(k, 2))

>>> 2
>>> 0

II. bisect.bisect_right、bisect.bisect

这两种都方法和bisect_left类似，唯一的区别就是出现相同元素会优先向右边插入，同样是返回索引值。

import bisect

if __name__ == '__main__':
    l = [1, 2, 3, 4]
    k = [1, 2, 3, 3, 3, 4]
    print(bisect.bisect_right(l, 3))
    print(bisect.bisect_right(k, 3))
    print(bisect.bisect(l, 3))
    print(bisect.bisect(k, 3))

>>> 3
>>> 5
>>> 3
>>> 5

III. bisect.insort_left(a, x, lo=0, hi=len(a))

不破壞 a 的排序的情况下下插入x，和 a.insert(bisect.bisect_left(a,x, lo, hi), x) 的效果相同。搜寻只需要 O(log n)时间而插入需要很慢的 O(n)时间，这使得插入操作主导了需要花费的时间。

if __name__ == '__main__':
    l = [1, 2, 3, 4]
    k = [1, 2, 3, 3, 3, 4]
    bisect.insort_right(l, 3)
    print(l)
    l.insert(bisect.bisect_right(l, 3), 3)
    print(l)

>>> [1, 2, 3, 3, 4]
>>> [1, 2, 3, 3, 3, 4]

IV. bisect.insort_right(a,x, lo=0, hi=len(a))、 bisect.insort(a, x,lo=0, hi=len(a))

与insort_left用法类似，区别在于对于重复的值会执行右插，插入的位置在所有a列表中的x的后面（右边）

if __name__ == '__main__':
    l = [1, 2, 3, 4]
    k = [1, 2, 3, 3, 3, 4]
    bisect.insort_right(l, 3)
    print(l)
    l.insert(bisect.bisect_right(l, 3), 3)
    print(l)

>>> [1, 2, 3, 3, 4]
>>> [1, 2, 3, 3, 3, 4]

2. 二分查找

前文说到过二分查找，二分查找也称折半查找（Binary Search），它是一种效率较高的查找方法。但是，折半查找要求线性表必须采用顺序存储结构，而且表中元素按关键字有序排列。每次查询会将表分为两部分，然后对于符合要求的那一部分继续执行同上所述的查找流程，直至无法再细分子表。

I.表必须采用顺序存储结构
II.必须按关键字大小有序排列

我们也可以自己来实现一下二分查找方法，虽然效率上会低于bisect

I. 循环实现

def binary_search_by_cycle(lst, item, start, end):
    while start <= end:
        mid = (start + end) // 2
        if lst[mid] == item:
            return mid
        elif lst[mid] < item:
            start = mid + 1
        else:
            end = mid - 1
    return None

if __name__ == '__main__':
    l = [i for i in range(1, 10000, 2)]
    print(binary_search_by_cycle(l, 9979, 0, len(l)))

>>> 4989

II. 递归实现

def binary_search_by_recursion(lst, item, start, end):
    if start > end:
        return None
    mid = (start + end) // 2
    if lst[mid] < item:
        return binary_search_by_recursion(lst, item, mid+1, end)
    elif lst[mid] > item:
        return binary_search_by_recursion(lst, item, start, mid-1)
    else:
        return mid

if __name__ == '__main__':
    l = [i for i in range(1, 10000, 2)]
    print(binary_search_by_recursion(l, 9979, 0, len(l)))

>>> 4989