给定一个长度为 N 的数列,A1,A2,…AN,如果其中一段连续的子序列 Ai,Ai+1,…Aj 之和是 K 的倍数,我们就称这个区间 [i,j] 是 K 倍区间。
你能求出数列中总共有多少个 K 倍区间吗?
输入格式
第一行包含两个整数 N 和 K。
以下 N 行每行包含一个整数 Ai。
输出格式
输出一个整数,代表 K 倍区间的数目。
数据范围
1≤N,K≤100000,
1≤Ai≤100000
输入样例:
5 2
1
2
3
4
5
输出样例:
6
代码如下:
#include <iostream>
#include <cstdio>
using namespace std;
int n, k;
int sum[100005], cnt[100005];
long long ans = 0;
int main() {
scanf("%d%d", &n, &k);
for (int i = 1; i <= n; i++) {
int x;
scanf("%d", &x);
sum[i] += (sum[i - 1] + x) % k;
ans += cnt[sum[i]];
cnt[sum[i]]++;
}
printf("%lld", ans + cnt[0]);
return 0;
}