牛客练习赛16 F - 选值

F

选值

题目描述 

给定n个数，从中选出三个数，使得最大的那个减最小的那个的值小于等于d，问有多少种选法。

输入描述:

第一行两个整数n，d(1 <= n <= 100,000，1 <= d <= 1000,000,000)；
第二行n个整数满足abs(ai) <= 1,000,000,000。数据保证a单调递增。

输出描述:

输出一个整数表示满足条件的选法。

示例1

输入

4 3
1 2 3 4

输出

4

示例2

输入

4 2
-3 -2 -1 0

输出

2

示例3

输入

5 19
1 10 20 30 50

输出

1

思路：其实这样想，c - a >= d  -> c - d <= a,这样我们枚举c，就可以找到向前找到一个最小的a,这样再选两个数a,b;就是组合问题啦。。。

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn = 1e5+7;
 
ll n,d;
ll a[maxn];
 
int main()
{
    scanf("%lld%lld",&n,&d);
    for(int i = 0; i < n; i++) scanf("%lld",&a[i]);
    ll sum = 0;
    for(int i = n - 1; i >= 1; i--)
    {
        int len = lower_bound(a,a+n,a[i] - d) - a;
        len = i - len;
        //cout << len << endl;
        ll l = (ll)len;
        if(len >= 2) {
            sum += l*(l - 1)/(ll)2;
        }
    }
    cout << sum << endl;
}