Python数据结构与算法---维护有序列表bisect

Posted 2021-06-28 一天一篇Python库

目录

• 前言
• 有序插入
• 重复值处理

前言

bisect实现了一个算法来向列表中插入元素，同时仍保持列表有序。

本篇，将详细介绍bisect库高效率的玩转列表。

有序插入

首先，我们来看看bisect库是如何实现列表的拆入的。具体代码如下所示：

import bisect

a = [7, 5, 4, 1, 9, 8, 2, 3, 6, 0, 5]
print(a)
new_a = []
for i in a:
    position = bisect.bisect(new_a, i)
    bisect.insort(new_a, i)
    print(position, new_a)

运行之后，效果如下：

可以看到，bisect会自动排序进行插入，position为插入的索引位置。当然，对于此类插入如果直接构建列表，然后排序，可能速度更快。不过这只仅仅对于短列表而言非常的快，对于非常长的列表而言，使用上面这种插入排序方式可以大大节省时间和内存，尤其是比较两个列表成员的操作需要开销很大的计算量时。

重复值处理

在实际的列表处理中，我们可能处理重复的值。如前文所示，多余的5是默认插入到重复值右边的，也就是说相当于使用insort_right()函数。同理，那么左边我们就可以用insort_left()函数。

import bisect

a = [7, 5, 4, 1, 9, 8, 2, 3, 6, 0, 5]
print(a)
new_a = []
for i in a:
    position = bisect.bisect_left(new_a, i)
    bisect.insort_left(new_a, i)
    print(position, new_a)

运行之后，效果如下：

读者可以对比一下上面两个图片，最后一行的索引变化。可以看到，一个是6，一个是5，因为我们主动变更，把重复值默认插入到左边了。