Submission #3791091

---

Source Code Expand

#include<bits/stdc++.h>
using namespace std;
#define ll long long
int main(){
  int n;
  cin>>n;
  ll a[n],s[n+1];
  s[0]=0;
  for(int i=0;i<n;i++){
    cin>>a[i];
    s[i+1]=a[i]+s[i];
  }
  sort(s,s+n+1);
  ll b=s[0],c=0,d=1;
  for(int i=1;i<=n;i++){
    if(b==s[i])
      d++;
    else{
      c+=d*(d-1)/2;
      d=1;
      b=s[i];
    }
  }
  c+=d*(d-1)/2;
  cout<<c;
}
    

Submission Info

Judge Result