CodeForces 895B XK Segments

最新推荐文章于 2022-07-22 13:33:59 发布

原创最新推荐文章于 2022-07-22 13:33:59 发布 · 622 阅读

0 ·

CC 4.0 BY-SA版权

ACM 专栏收录该内容

261 篇文章

订阅专栏

本文介绍了一道来自 CodeForces 的题目 895B 的解题思路与实现方法。该题要求找出所有可能的数对 (i, j)，使得在给定的序列中从 a[i] 到 a[j] 区间内恰好包含 k 个数为 x 的倍数。通过将序列排序并使用二分查找进行区间统计，实现了高效求解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：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;
}