小b喜欢和为K的倍数的序列。
现在有一个长度为n的序列A,请问A有多少个非空连续子序列是小b喜欢的。
输入
第一行输入一个正整数n;
第二行输入n个整数,表示A[i],以空格隔开;
第三行输入一个正整数K;
其中1≤n≤30000,对于任意A[i]有-10000≤A[i]≤10000,2≤K≤10000
输出
输出一个数,表示子序列的数目
输入样例
6
4 5 0 -2 -3 1
5
输出样例
7
前缀和的时候会超范围,也要进行处理
#include<bits/stdc++.h>
using namespace std;
int a[30010];
int main()
{
ios::sync_with_stdio(false);
int n,tmp,k;
long long sum=0;
cin>>n;
a[0]=0;
for(int i=1;i<=n;i++)
cin>>a[i];
cin>>k;
for(int i=1;i<=n;i++)
a[i]=(a[i-1]+a[i])%k;
for(int i=1;i<=n;i++)
{
for(int j=0;j<i;j++)
if((a[i]-a[j])%k==0)
sum++;
}
cout<<sum;
return 0;
}