CodeForces 895B XK Segments

题目链接：http://codeforces.com/problemset/problem/895/B

题意：给定n个数，从这些数中找到一个序列(i,j)，使得[a[i],a[j]]中恰好有k个数是x的倍数，求一共有多少个序列（PS：对于序列来说i，j的关系是无序的，也就是说可以i<j || i >= j）

思路：把n个数从小到大排序，然后求出大于等于a[i]的第一个x的倍数p，然后只需二分查找计算出p+k*x 和 max（a[i],p+(k-1)*x）之间数的个数求和即可。

 

#include <bits/stdc++.h>
#define LL long long
using namespace std;

int n;
LL x,k,a[110000];

int main(){
    scanf("%d%I64d%I64d",&n,&x,&k);
    for(int i = 0; i < n; i ++) scanf("%I64d",&a[i]);
    sort(a,a+n);
    LL sum = 0;
    for(int i = 0; i < n; i ++){
        LL p = ceil(a[i]*1.0/x)*x;
        sum += lower_bound(a,a+n,p+k*x) - lower_bound(a,a+n,max(a[i],p+(k-1)*x));
    }
    printf("%I64d\n",sum);
    return 0;
}