1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
#include <bits/stdc++.h>
#define PB push_back
using namespace std;

long long SUM3(vector<int>& arr){
    sort(arr.begin(), arr.end());
    int n=arr.size();
    long long res=0;
    for(int i=0; i<n; i++){
        int act=arr[i];
        int l=i+1;
        int r=n-1;
        while(l<r){
            int sum=arr[i]+arr[l]+arr[r];
            if(sum==0){
                for(int k=l; arr[k]==arr[l] && k!=r; k++){
                    res++;
                }
                r--;
            }
            else if(sum<0){
                l++;
            }
            else{
                r--;
            }
        }
    }
    return res;
}

int main(){
    ios_base::sync_with_stdio(NULL);
    cin.tie(NULL);

    int n;
    cin >> n;
    vector<int> arr(n);
    vector<int> bett;
    vector<int> pref(n+1);
    for(int i=0; i<n; i++){
        cin >> arr[i];
    }
    for(int i=0; i<n; i++){
        pref[i+1]=pref[i]+arr[i];
    }
    for(int i=0; i<n; i++){
        for(int j=i+1; j<n+1; j++){
            bett.PB(pref[j]-pref[i]);
        }
    }
    cout << SUM3(bett) << "\n";

    return 0;
}