和为k的倍数

試著忘記壹切 提交于 2019-12-01 10:17:55

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

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

#include<cstdio>
#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=3e4+10;
int a[N],sum[N],vis[N];
int main(){
    int n,k;
    cin>>n;
    for(int i=1;i<=n;i++)
    scanf("%d",&a[i]);
    cin>>k;
    int ans=0;
    vis[0]++;
    for(int i=1;i<=n;i++){
        sum[i]=(sum[i-1]+a[i]+k*10000)%k;
        ans+=vis[sum[i]];
        vis[sum[i]]++;
    }
    cout<<ans<<endl;
}
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!