小A的糖果
时间限制(普通/Java) : 1000 MS/ 3000 MS 运行内存限制 : 65536 KByte
描述
小A有n堆糖果,按照顺序排放,每一堆分别有 a1,a2,a3...an个,他
想把其中的一些连续的堆取出来,分给k个小伙伴,小A很公平,他不想有的人分的多,
有的人分的少,现在小A想知道,有多少种不同的取法。 1<=n<=100000,1<=ai<=100000, k<=100000
输入
第一行 两个整数 n和k;
接下来 n行,每行一个正整数 ai
输出
所求结果
样例输入
4 2
1 2 3 4
样例输出
4
题目来源
NUPT
#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#include<map>
#include<cmath>
#include<vector>
#include<algorithm>
#include<set>
#define mem(x,y) memset(x,y,sizeof(x))
#define scan(x) scanf("%d",&x)
#define scan2(x,y) scanf("%d%d",&x,&y)
#define eps 1e-8;
#define INF 100000000
#define MAXN 100005
using namespace std;
typedef long long ll;
int a[MAXN];
ll sum[MAXN];
ll cnt[MAXN];
int main()
{
int n,k;
ll ans=0;
cin>>n>>k;
mem(sum,0);
mem(cnt,0);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
cnt[0]=1;
for(int i=1;i<=n;i++)
{
sum[i]+=sum[i-1]+a[i];
int x;
x=sum[i]%k;
cnt[x]++;
ans+=cnt[x]-1;
}
cout<<ans<<endl;
return 0;
}