func_code_string: #include <bits/stdc++.h>   using namespace std;   int main() {     ios_base::sync_with_stdio(false);     cin.tie(NULL);     cout.tie(NULL);     ;     int t, n, i;     cin >> n;     int a[n];     for (i = 0; i < n; i++) cin >> a[i];     long long int ans = 0;     long long int pre[n + 1];     memset(pre, 0, sizeof(pre));     for (i = 1; i < n + 1; i++) pre[i] = pre[i - 1] + a[i - 1];     set<long long int> s;     int start = -1;     for (i = 0; i < n; i++) {       if (start < i) {         start = i;         s.insert(pre[i]);       }       while (start < n && !s.count(pre[start + 1])) {         start++;         s.insert(pre[start]);       }       s.erase(pre[i]);       ans += start - i;     }     cout << ans << endl;   }   