CodeForces 251 A.Points on Line（二分）

Description

给出数轴上$n$个点的坐标，问有多少种方案可以从中选取三个点使得其中距离最远点对距离不超过$d$

Input

第一行两个整数$n,d$，之后升序的输入$n$个点的坐标$a_i$$(1\le n\le 10^5,1\le d\le 10^9,|a_i|\le 10^9)$

Output

输出满足条件的方案数

Sample Input

4 3
1 2 3 4

Sample Output

1

Solution

枚举第一个点，之后二分找到距其距离不超过$d$的点数$cnt$，从中任意取两个均可

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100001;
int n,d,a[maxn];
ll C(int n)
{
    return (ll)n*(n-1)/2;
}
int main()
{
    while(~scanf("%d%d",&n,&d))
    {
        for(int i=0;i<n;i++)scanf("%d",&a[i]);
        ll ans=0;
        for(int i=0;i<n;i++)
        {
            int pos=upper_bound(a,a+n,a[i]+d)-a-1;
            ans+=C(pos-i);
        }
        printf("%I64d\n",ans);
    }
    return 0;
}