是Python太慢了吗
  • 板块P1102 A-B 数对
  • 楼主lixiao189
  • 当前回复10
  • 已保存回复10
  • 发布时间2020/8/21 16:11
  • 上次更新2023/11/6 19:45:15
查看原帖
是Python太慢了吗
59464
lixiao189楼主2020/8/21 16:11

用二分做这个题目,O(nlog(n))O(n * log(n)) 为毛被卡了,是Python太慢了吗

Python代码

def find(tmpArr, k, l): # k是要查找的数据 l是数组长度
    i = 0
    j = l - 1
    while i <= j:
        mid = int((i + j) / 2)
        if (arr[mid] > k):
            i = mid + 1
        else:
            j = mid - 1
    ansL = i
    i = 0
    j = l - 1
    while i <= j:
        mid = int((i + j) / 2)
        if (arr[mid] >= k):
            i = mid + 1
        else:
            j = mid - 1
    ansR = j
    return ansR - (ansL - 1)


ans = 0
n, c = map(int, input().split(" "))
arr = list(map(int, input().split(" ")))
arr.sort(reverse=True)  # 二分前对数组排序
for i in range(0, n):
    key = arr[i] - c
    ans = ans + find(arr, key, n)
print(ans)

2020/8/21 16:11
加载中...