题意:求出任意三个,其中这三个中最大的和最小的不超过d的组数,很巧的是这道题是从小到大输入的,其实对于一个数n,找到不大于n+d的最大的数,确定这两个之间的个数,然后以n作为第一个,然后在长度为l的任意找出2个,就是组合,二分求上界突然忘了,翻书就看到了这个函数upper_bound(),函数的返回值是小于等于n的最后一个元素的下一个,所以要-1
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int MAXN = 100005;
int n;
long long arr[MAXN],d;
int main(){
while (scanf("%d%lld",&n,&d) != EOF){
for (int i = 0; i < n; i++)
scanf("%lld",&arr[i]);
long long ans = 0;
for (int i = 0; i < n-2; i++){
int pos = upper_bound(arr+i,arr+n,arr[i]+d) - arr;
pos--;
ans += (pos-i)*(pos-i-1)/2;
}
printf("%lld\n",ans);
}
return 0;
}