小A的糖果

  

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