给定一个长度为 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<algorithm>
using namespace std;
long long aa[1234567];
long long ss[1234567];
int main()
{
int n,m;
cin>>n>>m;
long long res=0;
for(int i=1;i<=n;i++) cin>>aa[i];
for(int i=1;i<=n;i++)
{
aa[i]=(aa[i]+aa[i-1])%m;
ss[aa[i]%m]++;
}
res+=ss[0];
for(int i=0;i<m;i++)
res+=ss[i]*(ss[i]-1)/2;
cout<<res<<endl;
return 0;
}