2522 和为K的倍数

2522 和为K的倍数

1. 2 秒
2. 262,144 KB
3. 10 分
4. 2 级题

 

小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],b[30010],c[30010];
int main()
{
    int n,s,sum=0;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        b[i]=b[i-1]+a[i];
    }
    scanf("%d",&s);
    for(int i=0;i<=n;i++)
    {
        c[(b[i]%s+s)%s]++;
    }
    for(int i=0;i<s;i++)
    {
        sum+=c[i]*(c[i]-1)/2;
    }
    printf("%d\n",sum);
    return 0;
}