Python Bisect

简介: Python Bisect

bisect是python内置模块,用于有序序列的插入和查找


  • 查找: bisect(array, item)
  • import bisect
  • 若数组a是升序序列


使用总结:


bisect.bisect(a,c):返回数组a中最后一个<=c的索引值+1


若c存在于数组a,bisect.bisect_left(a,c)返回数组a中第一个出现c的索引值


若c不存在于数组a,bisect.bisect_left与bisect.bisect等效


若c存在于数组a,bisect.bisect_right(a,c)返回数组a中最后一个等于c的索引值+1


若c不存在于数组a,bisect.bisect_right与bisect.bisect等效


在查询一个有序的数组中有多少个>(<) c的时候,这个函数效率就很高


import bisect
n=int(input())
a=list(map(int,input().split()))
b=list(map(int,input().split()))
c=list(map(int,input().split()))
a.sort()
c.sort()
ans = 0
for i in range(n):
  x=bisect.bisect_left(a,b[i])
  y=len(c)-bisect.bisect_right(c,b[i])
  ans += x*y
print(ans)


image.png


这个代码是可以解决蓝桥杯一道真题:三元递增组,相较于用前缀和,代码量大大化简


前缀和代码:

n=int(input())
a=list(map(int,input().split()))
b=list(map(int,input().split()))
c=list(map(int,input().split()))
ans=[0 for i in range(max(a)+1)]
for i in a:
    ans[i]+=1
x=[0 for i in range(max(a)+1)]
x[0]=a.count(0)
for j in range(1,max(a)+1):
    x[j]=x[j-1]+ans[j]
cns=[0 for i in range(max(c)+1)]
for i in c:
    cns[i]+=1
y=[0 for i in range(max(c)+1)]
y[0]=c.count(0)
for j in range(1,max(c)+1):
    y[j]=y[j-1]+cns[j]
cnt=0
for k in range(n):
    if b[k]-1>len(x)-1:
        p=x[-1]
    elif b[k]-1<0:
        p=0
    else:
        p=x[b[k]-1]
    if len(y)-1<=b[k]:
        q=0
    else:
        q=y[-1]-y[b[k]]
    cnt+=p*q
print(cnt)


掌握一些内置的高效函数,大有裨益!


目录
相关文章
|
9月前
|
索引
python-enumerate()
python-enumerate
46 1
|
Python
python:函数用法
python:函数用法
46 0
|
4月前
|
存储 安全 Java
Python Tuples详解!
本文详细介绍了Python中的元组(Tuples),包括创建方式、操作方法及应用场景。元组是一种不可变的有序集合,支持多种数据类型。文章演示了如何创建空元组、使用字符串、列表及混合数据类型创建元组,并展示了元组的连接、切片、删除等操作。此外,还对比了元组与列表的区别,强调了元组的不可变性和高效性。最后,列举了元组的常见内置方法和函数,如`index()`、`count()`、`len()`等,并提供了实际应用示例。
63 2
|
7月前
|
Python
【Python】 已解决:NameError: name ‘python‘ is not defined
【Python】 已解决:NameError: name ‘python‘ is not defined
793 8
|
9月前
|
算法 开发者 索引
深入理解Python中的二分查找与bisect模块
深入理解Python中的二分查找与bisect模块
|
9月前
|
存储 C# Python
Python 教程之控制流(5)Python 中的 range() 与 xrange()
Python 教程之控制流(5)Python 中的 range() 与 xrange()
50 0
|
存储 索引 Python
Python|NumPy的常用操作
Python|NumPy的常用操作
|
索引 Python
Python Bisect
Python Bisect
123 0
Python Bisect
|
Python
Python循环器-itertools
Python循环器-itertools
112 0
Python循环器-itertools
|
索引 Python 容器
让你的Python代码更加Pythonic
让你的Python代码更加Pythonic
134 0