//Jakub Nowak XIV LO Wroclaw
#include<bits/stdc++.h>
using namespace std;
#define boost ios_base::sync_with_stdio(false); cin.tie(0);
#define int long long
#define vi vector<int>
#define pii pair<int,int>
#define pb push_back
#define st first
#define nd second
const int inf = (int)(1e18)+7;
void wypisz(auto &X) {
for(auto &x : X) cout << x << " ";
cout << "\n";
}
void wypisz_pair(auto &X) {
for(auto &x : X) cout << x.st << "|" << x.nd << " ";
cout << "\n";
}
void solve() {
int n;
vi A(1, 0);
vector<signed> PRE(1, 0);
int ile_1 = 0;//ile trojek o 1 roznym elemencie (wszystkie te same)
int ile_2 = 0;//ile trojek o 2 roznych elementach
int ile_123 = 0;//ile trojek o 1, 2 lub 3 roznych elementach
int ans = 0;
cin >> n;
for(int i=1; i<=n; i++) {
int x;
cin >> x;
A.pb(x);
}
//return;
for(int i=1; i<=n; i++) PRE.pb(PRE.back() + A[i]);
{//Zliczanie ile_123
//unordered_map<int, int> M;
#undef int
vi M(6e7+7, 0);
int s = signed(3e7 +2);
for(int j1 = n; j1 >= 1; j1--) {
for(int k1 = 1; k1 <=n; k1++) for(int k2 = k1; k2 <= n; k2++) M[PRE[k2]-PRE[k1-1]+ PRE[j1] + s]++;
for(int i1 = 1; i1 <= n; i1++) for(int i2 = i1; i2 <= n; i2++) ile_123 += M[-(PRE[i2] - PRE[i1-1] - PRE[j1-1]) + s];
}
#define int long long
}
//return;
unordered_map<int, int> M2;
for(int i = 1; i <= n; i++) for(int j = i; j <= n; j++) M2[PRE[j] - PRE[i-1]]++;
//return;
{//Zliczanie ile_2
for(int i = 1; i <= n; i++) for(int j = i; j <= n; j++) {
M2[PRE[j] - PRE[i-1]]--;
ile_2 += M2[-2*(PRE[j] - PRE[i-1])];
M2[PRE[j] - PRE[i-1]]++;
}
ile_2*=3;
}
{//Zliczanie ile_1
ile_1 = M2[0];
}
ans = (ile_123 - ile_2 - ile_1)/6;
cout << ans << "\n";
}
/*void gen() {
int n = 500;
cout << n << "\n";
for(int i=1; i<=n; i++) cout << (i%2 ? 20000 : -20000) << " ";
}*/
int32_t main() {
boost
//gen();
solve();
}
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 | //Jakub Nowak XIV LO Wroclaw #include<bits/stdc++.h> using namespace std; #define boost ios_base::sync_with_stdio(false); cin.tie(0); #define int long long #define vi vector<int> #define pii pair<int,int> #define pb push_back #define st first #define nd second const int inf = (int)(1e18)+7; void wypisz(auto &X) { for(auto &x : X) cout << x << " "; cout << "\n"; } void wypisz_pair(auto &X) { for(auto &x : X) cout << x.st << "|" << x.nd << " "; cout << "\n"; } void solve() { int n; vi A(1, 0); vector<signed> PRE(1, 0); int ile_1 = 0;//ile trojek o 1 roznym elemencie (wszystkie te same) int ile_2 = 0;//ile trojek o 2 roznych elementach int ile_123 = 0;//ile trojek o 1, 2 lub 3 roznych elementach int ans = 0; cin >> n; for(int i=1; i<=n; i++) { int x; cin >> x; A.pb(x); } //return; for(int i=1; i<=n; i++) PRE.pb(PRE.back() + A[i]); {//Zliczanie ile_123 //unordered_map<int, int> M; #undef int vi M(6e7+7, 0); int s = signed(3e7 +2); for(int j1 = n; j1 >= 1; j1--) { for(int k1 = 1; k1 <=n; k1++) for(int k2 = k1; k2 <= n; k2++) M[PRE[k2]-PRE[k1-1]+ PRE[j1] + s]++; for(int i1 = 1; i1 <= n; i1++) for(int i2 = i1; i2 <= n; i2++) ile_123 += M[-(PRE[i2] - PRE[i1-1] - PRE[j1-1]) + s]; } #define int long long } //return; unordered_map<int, int> M2; for(int i = 1; i <= n; i++) for(int j = i; j <= n; j++) M2[PRE[j] - PRE[i-1]]++; //return; {//Zliczanie ile_2 for(int i = 1; i <= n; i++) for(int j = i; j <= n; j++) { M2[PRE[j] - PRE[i-1]]--; ile_2 += M2[-2*(PRE[j] - PRE[i-1])]; M2[PRE[j] - PRE[i-1]]++; } ile_2*=3; } {//Zliczanie ile_1 ile_1 = M2[0]; } ans = (ile_123 - ile_2 - ile_1)/6; cout << ans << "\n"; } /*void gen() { int n = 500; cout << n << "\n"; for(int i=1; i<=n; i++) cout << (i%2 ? 20000 : -20000) << " "; }*/ int32_t main() { boost //gen(); solve(); } |
English