利用两个while把相等的区间找出来,然后计算区间内的数字数即可
#include <bits/stdc++.h>
using namespace std;
const int N = 2e5 + 5;
int a[N];
int main (){
int n , c; cin >> n >> c;
for(int i = 1 ; i <= n ; i ++) cin >> a[i];
sort(a + 1 , a + 1 + n);
long long ans = 0; //答案数量超过int,需要用long long
for(int i=1,j=1,k=1; i <= n ; i ++) {
while(j <= n && a[j] - a[i] < c ) j++; //用j、k查找数字相同的区间
while(k <= n && a[k] - a[i] <= c) k++; //区间[j,k]内所有数字相同
if(a[j]-a[i]==c && a[k-1]-a[i]==c && k-1>=1)
ans += k - j; //统计数对
}
cout << ans;
return 0;
}