蓝桥杯 PREV-40 历届试题 K倍区间

资源限制

时间限制：2.0s 内存限制：256.0MB

问题描述

给定一个长度为N的数列，A1, A2, … AN，如果其中一段连续的子序列Ai, Ai+1, … Aj(i <=
j)之和是K的倍数，我们就称这个区间[i, j]是K倍区间。
你能求出数列中总共有多少个K倍区间吗？

输入格式

第一行包含两个整数N和K。(1 <= N, K <= 100000) 　　
以下N行每行包含一个整数Ai。(1 <= Ai <= 100000)

输出格式

输出一个整数，代表K倍区间的数目。

从这个题的数据范围来看，要想得到满分，使用的算法复杂度应该为 O(n) 或者 O(n logn)。
第一次我观察规律之后，写了个双 for 循环，每次删除数组的第一个数，求前缀和，只得了 28 分。看了 yxc 大佬的视频和讲解后明白了思路。
利用前缀和可求解。
(a[r] - a[l]) % k == 0 等价于 a[r]%k == a[l]%k
有了这个关系式，我们在读入时求前缀和，再写一个循环，从第一位开始枚举，检测 a[i] % k 出现过几次…详情看代码。
第16行为什么要将 cnt[0]置为1，如果我们遇到的第一个右端点对 k 取余为0，那 ans 就应该加一，所以 cnt[0] 应该为1.

AC代码如下：

#include <cstdio>
#include <iostream>
typedef long long LL;
using namespace std;

int n, k;
LL a[100010], cnt[100010];

int main() {
	scanf("%d %d", &n, &k);
	for(int i = 1; i <= n; i++)	{
		scanf("%lld", &a[i]);
		a[i] += a[i-1];
	}
	LL ans = 0;
	cnt[0] = 1;
	for(int i = 1; i <= n; i++) {
		ans += cnt[a[i] % k];
		cnt[a[i] % k]++;
	}
	printf("%lld", ans);
}