CodeForces - 251A Points on Line

题意:求出任意三个,其中这三个中最大的和最小的不超过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;
}



你可能感兴趣的:(CodeForces - 251A Points on Line)