AcWing 1230. K倍区间
题目描述
给定一个长度为 N N N 的数列, A 1 , A 2 , … A N A_1, A_2, … A_N A1,A2,…AN ,如果其中一段连续的子序列 A i , A i + 1 , … A j A_i, A_{i+1}, … A_j Ai,Ai+1,…Aj 之和是 K K K 的倍数,我们就称这个区间 [ i , j ] [i,j] [i,j] 是 K K K 倍区间。
你能求出数列中总共有多少个 K K K 倍区间吗?
输入格式
第一行包含两个整数 N N N 和 K K K。
以下 N N N 行每行包含一个整数 A i A_i Ai。
输出格式
输出一个整数,代表 K K K 倍区间的数目。
数据范围
1
≤
N
,
K
≤
100000
1≤N,K≤100000
1≤N,K≤100000,
1
≤
A
i
≤
100000
1≤A_i≤100000
1≤Ai≤100000
输入样例:
5 2
1
2
3
4
5
输出样例:
6
思路
求区间 [ l , r ] [l,r] [l,r] 的和是 k k k 的倍数的个数。
求区间和,我们可以通过 前缀和 来求出。
定义 sum[i]
表示第
1
1
1 个元素到第
i
i
i 个元素的和,那么 s[r] - s[l-1]
就是区间
[
l
,
r
]
[l,r]
[l,r] 的和。
若满足条件:区间
[
l
,
r
]
[l,r]
[l,r] 的和是k的倍数,即 (s[r] - s[l-1]) % k == 0
,等价于 s[r] % k == s[l-1] % k
。
说人话,这也就意味着:
如果
s[r] mod k
和s[l - 1] mod k
的余数相等,那么s[r] - s[l - 1]
的差值必然是 k k k 的倍数。比如:13 % 7 == 20 % 7,则 (20 - 7) % 7 == 0
那么题目就是要我们求 前缀和%k==0 的组合有多少种。
用 cnt[i]
存储目前为止前缀和相同的个数,
i
i
i 表示这个前缀和的值。
每次用 res
来递加 cnt[i]
相同的个数,前面有几个 前缀和的值 和 当前前缀和 相等,那么这个前缀和就能和前面每一个组成一个组合,所以要 res += cnt[s[i]]
,然后再加上现在的前缀和,即 cnt[s[i]]++
。
初始化 cnt[0] = 1
,因为当 s[i] == 0
时,这个前缀和本身就是
k
k
k 的倍数,不需要再跟别的前缀和组合,计算结果时就要加上这一个。
代码
#include <iostream>
#include <cstring>
#include <algorithm>
#define ll long long
using namespace std;
const int N = 1e5 + 10;
int n, k;
ll s[N];
ll cnt[N];
int main()
{
cin >> n >> k;
ll res = 0;
cnt[0] = 1;
for (int i = 1; i <= n; i++){
cin >> s[i];
s[i] = (s[i] + s[i - 1]) % k; //每次前缀和都取模
res += cnt[s[i]]; //和前面每一个都组合一下
cnt[s[i]]++; //现在又多了一个
}
cout << res << endl;
return 0;
}