2522 和为K的倍数

佚名 6年前 (2019-04-15) 随笔 1205人围观抢沙发百度已收录

2522 和为K的倍数

2 秒
262,144 KB
10 分
2 级题

小b喜欢和为K的倍数的序列。

现在有一个长度为n的序列A，请问A有多少个非空连续子序列是小b喜欢的。

SRE实战互联网时代守护先锋，助力企业售后服务体系运筹帷幄！一键直达领取阿里云限量特价优惠。

收起

输入

第一行输入一个正整数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

输出样例

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