当三个国家中的任何两国实力之和都大于第三国的时候,这三个国家互相结盟就呈“三足鼎立”之势,这种状态是最稳定的。现已知本国的实力值,又给出 n 个其他国家的实力值。我们需要从这 n 个国家中找 2 个结盟,以成三足鼎立。有多少种选择呢?
输入格式:
输入首先在第一行给出 2 个正整数 n(2≤n≤105)和 P(≤10e9),分别为其他国家的个数、以及本国的实力值。随后一行给出 n 个正整数,表示n 个其他国家的实力值。每个数值不超过 10e9,数字间以空格分隔。
输出格式:
在一行中输出本国结盟选择的个数。
输入样例:
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)