Submission #2850718

---

Source Code Expand

#include<iostream>
#include<algorithm>
using namespace std;
int main (){
    long long n;
    cin >> n;
    long long a[n];
    long long ans=0;
    long long b[n];
    long long t=0;
    for(long long i=0;i<n;i++){
        cin >> a[i];
        if(i==0){
            b[i]=a[i];
        }else{
            b[i]=b[i-1]+a[i];
        }
        if(b[i]==0){
            ans++;
        }
    }
    sort(b, b+n);
    for(long long i=0;i<n;i++){
        if(i==0){
            continue;
        }
        if(b[i]==b[i-1]){
            t++;
            ans+=t;
        }else{
            t=0;
        }
    }
    cout << ans << endl;
    return 0;
}

Submission Info

Judge Result