2522 和为K的倍数

时间:2019-04-15
本文章向大家介绍2522 和为K的倍数,主要包括2522 和为K的倍数使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

2522 和为K的倍数

 

小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;
}