欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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)

 

 

相关标签: python3