#include<bits/stdc++.h> #define FOR(i,a,b) for(int i=a;i<(int)b;++i) #define FORD(i,a,b) for(int i=a;i>=(int)b;--i) #define PB push_back #define EB emplace_back #define FI first #define SE second #define umap unordered_map #define uset unordered_set #define vi vector<int> #define vvi vector<vi> #define vll vector<ll> #define vvll vector<vll> #define vpii vector<pii> #define pii pair<int, int> #define pll pair<ll, ll> #define ALL(X) (X).begin(),(X).end() #ifndef DEBUG #define endl (char)10 #endif using namespace std; using ll = long long; using ld = long double; template <class T> istream& operator>> (istream& is, vector<T>& vec){ FOR(i,0,vec.size()) is >> vec[i]; return is; } template <class T> ostream& operator<< (ostream& os, vector<T>& vec){ for(auto& t : vec) os << t << " "; return os; } template<class T, class U> ostream& operator<< (ostream& os, const pair<T, U>& p){ os << p.FI << " " << p.SE; return os; } template<class T, class U> istream& operator>> (istream& is, pair<T, U>& p){ is >> p.FI >> p.SE; return is; } const int MS = 10000000; const int SS = 2 * MS + 5; int main () { ios_base::sync_with_stdio(false); cin.tie(0); int n; cin >> n; vi VV(n); cin >> VV; vi co; vi licz(SS, 0); FOR(i,0,n) { int s = 0; FOR(j,i,n){ s += VV[j]; co.PB(s); licz[s + MS]++; } } sort(ALL(co)); co.erase(unique(ALL(co)), co.end()); ll ans = 0; FOR(i,0,co.size()){ int x = co[i]; ll mile = licz[x + MS]; if(mile > 1){ int v = -2 * x; int iv = v + MS; if (iv < 0) break; if (iv < SS) { if (licz[iv] > 0) { ll p = mile; p *= p - 1; p /= 2; if (x == 0) ans += (p * (mile - 2)) / 3; else ans += p * licz[iv]; } } } FOR(j,i+1,co.size()){ int v = -(co[i] + co[j]); if (v <= co[j]) break; int iv = v + MS; if (iv < SS) { if (licz[iv] > 0) { ll p = licz[co[i] + MS]; p *= licz[co[j] + MS]; ans += p * licz[iv]; } } } } cout << ans << endl; }
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 | #include<bits/stdc++.h> #define FOR(i,a,b) for(int i=a;i<(int)b;++i) #define FORD(i,a,b) for(int i=a;i>=(int)b;--i) #define PB push_back #define EB emplace_back #define FI first #define SE second #define umap unordered_map #define uset unordered_set #define vi vector<int> #define vvi vector<vi> #define vll vector<ll> #define vvll vector<vll> #define vpii vector<pii> #define pii pair<int, int> #define pll pair<ll, ll> #define ALL(X) (X).begin(),(X).end() #ifndef DEBUG #define endl (char)10 #endif using namespace std; using ll = long long; using ld = long double; template <class T> istream& operator>> (istream& is, vector<T>& vec){ FOR(i,0,vec.size()) is >> vec[i]; return is; } template <class T> ostream& operator<< (ostream& os, vector<T>& vec){ for(auto& t : vec) os << t << " "; return os; } template<class T, class U> ostream& operator<< (ostream& os, const pair<T, U>& p){ os << p.FI << " " << p.SE; return os; } template<class T, class U> istream& operator>> (istream& is, pair<T, U>& p){ is >> p.FI >> p.SE; return is; } const int MS = 10000000; const int SS = 2 * MS + 5; int main () { ios_base::sync_with_stdio(false); cin.tie(0); int n; cin >> n; vi VV(n); cin >> VV; vi co; vi licz(SS, 0); FOR(i,0,n) { int s = 0; FOR(j,i,n){ s += VV[j]; co.PB(s); licz[s + MS]++; } } sort(ALL(co)); co.erase(unique(ALL(co)), co.end()); ll ans = 0; FOR(i,0,co.size()){ int x = co[i]; ll mile = licz[x + MS]; if(mile > 1){ int v = -2 * x; int iv = v + MS; if (iv < 0) break; if (iv < SS) { if (licz[iv] > 0) { ll p = mile; p *= p - 1; p /= 2; if (x == 0) ans += (p * (mile - 2)) / 3; else ans += p * licz[iv]; } } } FOR(j,i+1,co.size()){ int v = -(co[i] + co[j]); if (v <= co[j]) break; int iv = v + MS; if (iv < SS) { if (licz[iv] > 0) { ll p = licz[co[i] + MS]; p *= licz[co[j] + MS]; ans += p * licz[iv]; } } } } cout << ans << endl; } |