python中二分查找与插入模块-bisect
程序员文章站
2024-03-17 19:21:22
...
python中有一个用于二分查找与插入的bisect模块,可以用来寻找列表中第一个大于或者等于目标值的数组下标,其中bisect模块包含几个方法,并且方法中可以传递四个参数,例如模块中的bisect_left方法
bisect.bisect_left(a, x, lo=0, hi=len(a))
第一个参数表示查找的有序列表,第二个参数表示待查找的目标值,第三、四个参数表示列表中查找的范围,不写默认是查找的是列表的整个范围
bisect_left方法对于查找目标值x,若x存在于有序列表中那么返回列表中x第一次出现的位置,若列表中没有返回列表中第一个大于x的位置,bisect_right方法用来返回列表中第一个大于x的位置,除了二分查找目标值x之外,还可以使用使用insort_left、insort_right方法往列表中插入一个数字,并且使得插入数字x之后的列表有序
下面是具体的例子:
import bisect
if __name__ == '__main__':
a = [1, 2, 3, 3, 6, 9, 10, 20]
# bisect_left: 二分查找第一个大于等于目标值x的下标
print(bisect.bisect_left(a, 2))
# bisect_right: 二分查找第一个大于目标值x的下标
print(bisect.bisect_right(a, 3))
print(bisect.bisect_left(a, 7))
print(bisect.bisect_right(a, 8))
# 往列表a插入一个数字3并且使得列表有序
bisect.insort_left(a, 3)
print(a)
# 往列表a插入一个数字3并且使得列表有序
bisect.insort_right(a, 4)
print(a)