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
56
57
58
59
60
61
62
63
64
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;


int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(nullptr);
	int n;
	cin>>n;
	vector<int>d(n);
	vector<int>z(n+1);
	for(int i=0; i<n; i++){
		cin>>d[i];
	}
	for(int i=1; i<n+1; i++){
		z[i] = d[i-1] + z[i-1];
	}
	vector<int> vv;
	vector<pair<int, int>> v;
	for(int i=1; i<=n; i++){
		for(int j=0; j<i; j++){
			vv.push_back(z[i] - z[j]);
		}
	}
	sort(vv.begin(), vv.end());
	v.push_back({vv[0], 1});
	for(unsigned i=1; i<vv.size(); i++){
		if(vv[i] == v.back().first){
			v.back().second++;
		}else{
			v.push_back({vv[i], 1});
		}
	}
	LL res = 0;
	for(unsigned i=0; i<v.size(); i++){
		for(int j=i+1, k=v.size()-1; j < k;){
			if(v[i].first + v[j].first + v[k].first == 0){
				res += v[i].second * v[j].second * v[k].second;
				j++;
			}else if(v[i].first + v[j].first + v[k].first > 0){
				k--;
			}else{
				j++;
			}
		}
	}
	for(auto [a, b] : v){
		if(a == 0){
			if(b == 3){
				res++;
			}else if(b == 4){
				res += 4;
			}else if(b == 5){
				res += 10;
			}else if(b > 5) {
				res += b*(b-1)*(b-2)/6;
			}
			break;
		}
	}
	cout<<res<<"\n";
	return 0;
}