题目描述
AcWing1230. K倍区间(前缀和+同余)
给定一个长度为N的数列,
A
1
,
A
2
,
.
.
.
A
N
A_1, A_2, ... A_N
A1,A2,...AN,如果其中一段连续的子序列
A
i
,
A
i
+
1
,
.
.
.
A
j
(
i
<
=
j
)
A_i, A_{i+1}, ... A_j(i <= j)
Ai,Ai+1,...Aj(i<=j)之和是K的倍数,我们就称这个区间[i, j]是K倍区间。
你能求出数列中总共有多少个K倍区间吗?
输入格式
第一行包含两个整数 N 和 K。
以下 N 行每行包含一个整数 Ai。
5 2
1
2
3
4
5
输出格式
输出一个整数,代表 K 倍区间的数目。
6
数据范围
1
≤
N
,
K
≤
100000
1≤N,K≤100000
1≤N,K≤100000,
1
≤
A
i
≤
100000
1≤A_i≤100000
1≤Ai≤100000
C++ (前缀和+同余)
#include <iostream>
using namespace std;
typedef long long LL;
const int N = 1e5 + 10;
int n, k;
LL s[N], cnt[N];
int main() {
scanf("%d%d", &n, &k);
for (int i = 1; i <= n;i++) {
scanf("%d", &s[i]);
s[i] += s[i-1];
}
LL res = 0;
cnt[0] = 1; // // 第一次遇到区间和为0时,应该直接+1,s[i]=0指的是从1到i直接就是满足条件的
for (int i = 1; i <= n; i++) {
res += cnt[s[i]%k]; // 这个 s[i]%k 可以和前面取模为 s[i]%k的分别构成一个k倍区间
// 所以此时又新增了cnt[s[i]%k]个K倍区间
cnt[s[i]%k]++;
}
cout << res << endl;
return 0;
}