#include <bits/stdc++.h>
#pragma GCC optimize("O3", "unroll-loops")
#define pb emplace_back
#define ins insert
#define mp make_pair
#define ssize(x) (int)x.size()
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
#define pii pair <int, int>
#define pll pair <ll, ll>
#define pld pair <ld, ld>
#define st first
#define nd second
using namespace std;
using ll = int_fast64_t;
// using lll = __int128_t;
using ld = long double;
const int oo = 1e9 + 7;
const ll ool = 1e18;
const int mod = 1e9 + 7;
void solve(){
int n; cin >> n;
vector <int> in(n + 1), pref(n + 1);
for(int i = 1; i <= n; i ++){
cin >> in[i];
pref[i] = pref[i - 1] + in[i];
}
vector <int> tab;
unordered_map <int, int> cnt;
for(int i = 1; i <= n; i ++){
for(int j = i; j <= n; j ++){
tab.pb(pref[j] - pref[i - 1]);
++ cnt[pref[j] - pref[i - 1]];
}
}
ll ans = 0;
for(int i = 0; i < ssize(tab); i ++){
for(int j = i + 1; j < ssize(tab); j ++){
int x = tab[i], y = tab[j];
if(x == - x - y && y == - x - y) ans += (cnt[- x - y] - 2);
else if(x == - x - y || y == - x - y) ans += (cnt[- x - y] - 1);
else ans += cnt[- x - y];
}
}
cout << ans / 3 << '\n';
}
signed main(){
ios_base::sync_with_stdio(0); cin.tie(0);
int t; t = 1;
// cin >> t;
while(t --) 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 | #include <bits/stdc++.h> #pragma GCC optimize("O3", "unroll-loops") #define pb emplace_back #define ins insert #define mp make_pair #define ssize(x) (int)x.size() #define all(x) x.begin(), x.end() #define rall(x) x.rbegin(), x.rend() #define pii pair <int, int> #define pll pair <ll, ll> #define pld pair <ld, ld> #define st first #define nd second using namespace std; using ll = int_fast64_t; // using lll = __int128_t; using ld = long double; const int oo = 1e9 + 7; const ll ool = 1e18; const int mod = 1e9 + 7; void solve(){ int n; cin >> n; vector <int> in(n + 1), pref(n + 1); for(int i = 1; i <= n; i ++){ cin >> in[i]; pref[i] = pref[i - 1] + in[i]; } vector <int> tab; unordered_map <int, int> cnt; for(int i = 1; i <= n; i ++){ for(int j = i; j <= n; j ++){ tab.pb(pref[j] - pref[i - 1]); ++ cnt[pref[j] - pref[i - 1]]; } } ll ans = 0; for(int i = 0; i < ssize(tab); i ++){ for(int j = i + 1; j < ssize(tab); j ++){ int x = tab[i], y = tab[j]; if(x == - x - y && y == - x - y) ans += (cnt[- x - y] - 2); else if(x == - x - y || y == - x - y) ans += (cnt[- x - y] - 1); else ans += cnt[- x - y]; } } cout << ans / 3 << '\n'; } signed main(){ ios_base::sync_with_stdio(0); cin.tie(0); int t; t = 1; // cin >> t; while(t --) solve(); return 0; } |
English