làm bằng đếm phân phối 
c++

$\color{#1fdb91}{\texttt{Mazh}}\color{#0aaec9}{\texttt{fan}}\color{#e9418e}{\texttt{Jack}}$

#include <bits/stdc++.h>
const int N=1e6+1;
using namespace std;
int n,a[N],s=0,dem=0;
unordered_map<int, int>d;
int main() {
    freopen("TONG.INP","r",stdin);
    freopen("TONG.OUT","w",stdout);
    cin>>n;
    for (int i=0;i<n;i++) {
        cin>>a[i];
    }
    for (int i=0;i<n;i++) {
        s+=a[i];
        if (s==0) dem++;
        dem+=d[s];

        d[s]++;
    }
    cout<<dem;
    return 0;
}

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 1e5;
int n, a[N+5];
long long f[N+5];
long long res=0;
map<long long,int>s;
int main()
{

    ios::sync_with_stdio(0);
    cin.tie(0), cout.tie(0);
    ios_base::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    freopen("TONG.inp", "r", stdin);
    freopen("TONG.out", "w", stdout);
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }f[0]=0;
    for(int i=1;i<=n;i++){
        f[i]=f[i-1]+a[i];
    }s[0]=1;
    for(int i=1;i<=n;i++){
        res=res+s[f[i]];
            s[f[i]]++;
    }
    cout<<res;
}