ZZUSOFTOJ-小光的难题：前缀和，同余定理

题目

AC代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define N 100000+10
ll add[N];
ll a[N],b[N];
ll cnt;
int main()
{
  ll n,mod;
  cin>>n>>mod;
  for(ll i=1;i<=n;i++) cin>>a[i];
  for(ll i=1;i<=n;i++)
  {
      add[i]+=add[i-1]+a[i];
      b[add[i]%mod]++;
  }
  cnt+=b[0];
  for(ll i=0;i<mod;i++)
  {
      cnt=cnt+(b[i]*(b[i]-1))/2;
  }
  cout<<cnt<<endl;
}