AcWing 1230. K倍区间-526互联

给定一个长度为 $N$ 的数列， $A_{1}, A_{2}, \dots A_{N}$ ，如果其中一段连续的子序列 $A_{i}, A_{i + 1}, \dots A_{j}$ 之和是 $K$ 的倍数，我们就称这个区间 $[i, j]$ 是 $K$ 倍区间。

你能求出数列中总共有多少个 $K$ 倍区间吗？

输入格式

第一行包含两个整数 $N$ 和 $K$ 。

以下 $N$ 行每行包含一个整数 $A_{i}$ 。

输出格式

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

数据范围

$1 \leq N, K \leq 100000$ ,
$1 \leq A_{i} \leq 100000$

输入样例：

输出样例：

代码实现：

#include<iostream>
using namespace std;
#define int long long
const int N=1e5+5;
//cnt[i]数组表示余数为i的序列的个数
int a[N],b[N],cnt[N];
signed main(){
	int n,k;
	cin>>n>>k;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		b[i]=a[i]+b[i-1];
	}
	//b[r]%k == b[l-1]%k
	//由于l从1开始，所以l-1从0开始，但是r从1开始
	int res=0;
	for(int i=0;i<=n;i++){
		res += cnt[b[i] % k];   
        cnt[b[i] % k] ++; 
	}
	cout<<res<<endl;
	return 0;
}