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 longusing 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;}