L2-2 三足鼎立PTA

当三个国家中的任何两国实力之和都大于第三国的时候，这三个国家互相结盟就呈“三足鼎立”之势，这种状态是最稳定的。

现已知本国的实力值，又给出 n 个其他国家的实力值。我们需要从这 n 个国家中找 2 个结盟，以成三足鼎立。有多少种选择呢？

输入格式：

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

输出格式：

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

输入样例：

7 30
42 16 2 51 92 27 35

输出样例：

9

样例解释：

能联合的另外 2 个国家的 9 种选择分别为：

{16, 27}, {16, 35}, {16, 42}, {27, 35}, {27, 42}, {27, 51}, {35, 42}, {35, 51}, {42, 51}。

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int n,p,a[N];
long long res;
int main(){
	scanf("%d%d",&n,&p);
	for(int i=0;i<n;i++) scanf("%d",&a[i]);
	sort(a,a+n);
	for(int i=0;i<n;i++){
		int x=upper_bound(a+i+1,a+n,abs(a[i]-p))-a;
		int y=lower_bound(a+i+1,a+n,a[i]+p)-a;
		res+=y-x;
	}
	cout<<res;
	return 0;
}