#pragma GCC optimize("O3") #include <bits/stdc++.h> using namespace std; #define FOR(i, n) for (int i = 0; i < n; ++i) #define f first #define s second #define pb push_back #define all(s) s.begin(), s.end() #define sz(s) (int)s.size() #define czas std::chrono::steady_clock::now() #define cnt std::chrono::duration_cast<std::chrono::microseconds>(czas - beg).count() * 1e-6 using vi = vector<int>; using ll = long long; using ull = unsigned long long; using pll = pair<ll, ll>; using pull = pair<ull, ull>; using vll = vector<ll>; using ii = pair<int, int>; const int MAX = 3 * 20000 * 500 + 6; int mapa[MAX * 2]; void solve() { int n; cin >> n; vi pref; pref.pb(0); for (int i = 0; i < n; i++) { int temp; cin >> temp; pref.pb(pref.back() + temp); } ll ans = 0; for (int sr = sz(pref) - 1; sr >= 0; sr--) { for (int pocz = 0; pocz < sz(pref); ++pocz) { for (int kon = pocz + 1; kon < sz(pref); ++kon) { // cout << "POCZ KON SR " << pocz << ' ' << kon << ' ' << sr << ' ' << mapa[pref[sr] + pref[pocz] - pref[kon]] << '\n'; ans += mapa[MAX + pref[sr] + pref[pocz] - pref[kon]]; } } for (int pocz = 0; pocz < sz(pref); ++pocz) { for (int kon = pocz + 1; kon < sz(pref); ++kon) { mapa[MAX + pref[kon] + pref[sr] - pref[pocz]]++; } } } unordered_map<int, int> ile; for (int i = 0; i < sz(pref); i++) { for (int j = i + 1; j < sz(pref); j++) { ile[pref[j] - pref[i]]++; } } for (auto & [val, quant] : ile) { if (val != 0) { if (ile.count(-2 * val)) { // cout << "uyyy " << quant << ' ' << ile[-2 * val] << '\n'; ans -= 1ll * 3 * quant * ile[-2 * val]; } } else { // cout << "zero " << quant << '\n'; ans -= quant; ans -= 1ll * 3 * quant * (quant - 1); } } cout << ans / 6 << '\n'; } int main() { ios::sync_with_stdio(0); cin.tie(0); int tests = 1; // cin >> tests; for (int test = 1; test <= tests; test++) { 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 | #pragma GCC optimize("O3") #include <bits/stdc++.h> using namespace std; #define FOR(i, n) for (int i = 0; i < n; ++i) #define f first #define s second #define pb push_back #define all(s) s.begin(), s.end() #define sz(s) (int)s.size() #define czas std::chrono::steady_clock::now() #define cnt std::chrono::duration_cast<std::chrono::microseconds>(czas - beg).count() * 1e-6 using vi = vector<int>; using ll = long long; using ull = unsigned long long; using pll = pair<ll, ll>; using pull = pair<ull, ull>; using vll = vector<ll>; using ii = pair<int, int>; const int MAX = 3 * 20000 * 500 + 6; int mapa[MAX * 2]; void solve() { int n; cin >> n; vi pref; pref.pb(0); for (int i = 0; i < n; i++) { int temp; cin >> temp; pref.pb(pref.back() + temp); } ll ans = 0; for (int sr = sz(pref) - 1; sr >= 0; sr--) { for (int pocz = 0; pocz < sz(pref); ++pocz) { for (int kon = pocz + 1; kon < sz(pref); ++kon) { // cout << "POCZ KON SR " << pocz << ' ' << kon << ' ' << sr << ' ' << mapa[pref[sr] + pref[pocz] - pref[kon]] << '\n'; ans += mapa[MAX + pref[sr] + pref[pocz] - pref[kon]]; } } for (int pocz = 0; pocz < sz(pref); ++pocz) { for (int kon = pocz + 1; kon < sz(pref); ++kon) { mapa[MAX + pref[kon] + pref[sr] - pref[pocz]]++; } } } unordered_map<int, int> ile; for (int i = 0; i < sz(pref); i++) { for (int j = i + 1; j < sz(pref); j++) { ile[pref[j] - pref[i]]++; } } for (auto & [val, quant] : ile) { if (val != 0) { if (ile.count(-2 * val)) { // cout << "uyyy " << quant << ' ' << ile[-2 * val] << '\n'; ans -= 1ll * 3 * quant * ile[-2 * val]; } } else { // cout << "zero " << quant << '\n'; ans -= quant; ans -= 1ll * 3 * quant * (quant - 1); } } cout << ans / 6 << '\n'; } int main() { ios::sync_with_stdio(0); cin.tie(0); int tests = 1; // cin >> tests; for (int test = 1; test <= tests; test++) { solve(); } return 0; } |