PTA-python-三足鼎立

当三个国家中的任何两国实力之和都大于第三国的时候，这三个国家互相结盟就呈“三足鼎立”之势，这种状态是最稳定的。现已知本国的实力值，又给出 n 个其他国家的实力值。我们需要从这 n 个国家中找 2 个结盟，以成三足鼎立。有多少种选择呢？

输入格式：
输入首先在第一行给出 2 个正整数 n（2≤n≤10​5）和 P（≤10e​9），分别为其他国家的个数、以及本国的实力值。随后一行给出 n 个正整数，表示n 个其他国家的实力值。每个数值不超过 10e​9，数字间以空格分隔。

输出格式：
在一行中输出本国结盟选择的个数。

输入样例：

7 30
42 16 2 51 92 27 35

输出样例：
9

代码：

import math
n,p=map(int,input().split())
a=[0]
s=0
A = input()
B = [int(n) for n in A.split()]
for i in range(1,n+1):
    a.append(B[i-1])
a.sort()
for i in range(1,n+1):
    l=i+1
    r=n
    minn=math.fabs(a[i]-p)
    maxx=a[i]+p
    if l>r:
        break
    while l<=r:
        mid=(l+r)//2
        if a[mid]<=minn:
            l=mid+1
        else:
            r=mid-1
    num1=l
    l=i+1
    r=n
    while l<=r:
        mid=(l+r)//2
        if a[mid]<maxx:
            l=mid+1
        else:
            r=mid-1
    num2=r
    s+=(num2-num1+1)
print(s)

点一个赞再走吧