#include <iostream>
#include <vector>
using namespace std;
#define MAX_VALUE 40000000
void input(int & n, vector<int> & UC) {
cin >> n;
UC.resize(n);
for(int i = 0; i < n; i++) {
cin >> UC[i];
}
}
void make_prefsum(int const n, vector<int> const & UC, vector<int> & prefsum) {
prefsum.resize(n+1);
prefsum[0] = 0;
for(int i = 0; i < n; i++) {
prefsum[i+1] = prefsum[i] + UC[i];
}
}
long long get_result(int const n, vector<int> const & prefsum) {
long long result = 0;
vector<int> count(MAX_VALUE*2+5, 0);
for(int b1 = n - 1; b1 >= 0; b1--) {
for(int a1 = b1 - 1; a1 >= 0; a1--) {
for(int a2 = n; a2 > a1; a2--) {
result += count[MAX_VALUE + prefsum[b1] + prefsum[a1] - prefsum[a2]];
}
}
for(int c1 = b1 + 1; c1 <= n-1; c1++) {
for(int c2 = c1 + 1; c2 <= n; c2++) {
count[MAX_VALUE + prefsum[b1] + prefsum[c2] - prefsum[c1]]++;
}
}
for(int b2 = b1 + 1; b2 <= n; b2++) {
for(int c2 = b1 + 1; c2 <= n; c2++) {
count[MAX_VALUE + prefsum[b2] + prefsum[c2] - prefsum[b1]]++;
}
}
for(int c2 = b1 + 1; c2 <= n; c2++) {
count[MAX_VALUE + prefsum[b1] + prefsum[c2] - prefsum[b1]]++;
}
}
fill(count.begin(), count.end(), 0);
for(int ab1 = n - 1; ab1 >= 0; ab1--) {
for(int a2 = ab1 + 1; a2 <= n; a2++) {
for(int b2 = a2 + 1; b2 <= n; b2++) {
result += count[MAX_VALUE + prefsum[ab1] + prefsum[ab1] - prefsum[a2] - prefsum[b2]];
}
}
for(int c2 = ab1 + 1; c2 <= n; c2++) {
count[MAX_VALUE + prefsum[c2] - prefsum[ab1]]++;
}
}
fill(count.begin(), count.end(), 0);
for(int bc1 = 0; bc1 < n; bc1++) {
for(int b2 = bc1 + 1; b2 <= n; b2++) {
for(int c2 = b2 + 1; c2 <= n; c2++) {
result += count[MAX_VALUE + prefsum[bc1] + prefsum[bc1] - prefsum[b2] - prefsum[c2]];
}
}
for(int a2 = bc1 + 1; a2 <= n; a2++) {
count[MAX_VALUE + prefsum[a2] - prefsum[bc1]]++;
}
}
fill(count.begin(), count.end(), 0);
for(int b2 = n; b2 >= 0; b2--) {
for(int a2 = b2 - 1; a2 >= 0; a2--) {
for(int abc1 = a2 - 1; abc1 >= 0; abc1--) {
result += count[MAX_VALUE + prefsum[abc1] + prefsum[abc1] + prefsum[abc1] - prefsum[a2] - prefsum[b2]];
}
}
count[MAX_VALUE + prefsum[b2]]++;
}
return result;
}
void solve() {
int n;
vector<int> UC;
input(n, UC);
vector<int> prefsum;
make_prefsum(n, UC, prefsum);
cout << get_result(n, prefsum) << "\n";
}
int main(){
ios_base::sync_with_stdio(0);
cin.tie(0);
solve();
return 0;
}
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 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 | #include <iostream> #include <vector> using namespace std; #define MAX_VALUE 40000000 void input(int & n, vector<int> & UC) { cin >> n; UC.resize(n); for(int i = 0; i < n; i++) { cin >> UC[i]; } } void make_prefsum(int const n, vector<int> const & UC, vector<int> & prefsum) { prefsum.resize(n+1); prefsum[0] = 0; for(int i = 0; i < n; i++) { prefsum[i+1] = prefsum[i] + UC[i]; } } long long get_result(int const n, vector<int> const & prefsum) { long long result = 0; vector<int> count(MAX_VALUE*2+5, 0); for(int b1 = n - 1; b1 >= 0; b1--) { for(int a1 = b1 - 1; a1 >= 0; a1--) { for(int a2 = n; a2 > a1; a2--) { result += count[MAX_VALUE + prefsum[b1] + prefsum[a1] - prefsum[a2]]; } } for(int c1 = b1 + 1; c1 <= n-1; c1++) { for(int c2 = c1 + 1; c2 <= n; c2++) { count[MAX_VALUE + prefsum[b1] + prefsum[c2] - prefsum[c1]]++; } } for(int b2 = b1 + 1; b2 <= n; b2++) { for(int c2 = b1 + 1; c2 <= n; c2++) { count[MAX_VALUE + prefsum[b2] + prefsum[c2] - prefsum[b1]]++; } } for(int c2 = b1 + 1; c2 <= n; c2++) { count[MAX_VALUE + prefsum[b1] + prefsum[c2] - prefsum[b1]]++; } } fill(count.begin(), count.end(), 0); for(int ab1 = n - 1; ab1 >= 0; ab1--) { for(int a2 = ab1 + 1; a2 <= n; a2++) { for(int b2 = a2 + 1; b2 <= n; b2++) { result += count[MAX_VALUE + prefsum[ab1] + prefsum[ab1] - prefsum[a2] - prefsum[b2]]; } } for(int c2 = ab1 + 1; c2 <= n; c2++) { count[MAX_VALUE + prefsum[c2] - prefsum[ab1]]++; } } fill(count.begin(), count.end(), 0); for(int bc1 = 0; bc1 < n; bc1++) { for(int b2 = bc1 + 1; b2 <= n; b2++) { for(int c2 = b2 + 1; c2 <= n; c2++) { result += count[MAX_VALUE + prefsum[bc1] + prefsum[bc1] - prefsum[b2] - prefsum[c2]]; } } for(int a2 = bc1 + 1; a2 <= n; a2++) { count[MAX_VALUE + prefsum[a2] - prefsum[bc1]]++; } } fill(count.begin(), count.end(), 0); for(int b2 = n; b2 >= 0; b2--) { for(int a2 = b2 - 1; a2 >= 0; a2--) { for(int abc1 = a2 - 1; abc1 >= 0; abc1--) { result += count[MAX_VALUE + prefsum[abc1] + prefsum[abc1] + prefsum[abc1] - prefsum[a2] - prefsum[b2]]; } } count[MAX_VALUE + prefsum[b2]]++; } return result; } void solve() { int n; vector<int> UC; input(n, UC); vector<int> prefsum; make_prefsum(n, UC, prefsum); cout << get_result(n, prefsum) << "\n"; } int main(){ ios_base::sync_with_stdio(0); cin.tie(0); solve(); return 0; } |
English