#include <bits/stdc++.h>
using namespace std;
#define FOR(i, l, r) for(int i = (l); i <= (r); ++i)
#define REP(i, n) FOR(i, 0, (n) - 1)
#define rep(i, a, b) for(int i = a; i < (b); ++i)
#define all(x) x.begin(), x.end()
#define sz(x) (int)(x).size()
#define ssize(x) int(x.size())
#define pb push_back
#define fi first
#define se second
#define ld long double
typedef long long ll;
typedef vector<int> vi;
typedef pair<int, int> pii;
typedef pair <ld, ld> pd;
using LL=long long;
#ifdef DEBUG
auto &operator<<(auto &o, pair<auto, auto> p) {
return o << "()" << p.first << ", " << p.second <<")";
}
auto operator<<(auto &o, auto x)-> decltype(x.end(), o) {
o << "{";int i = 0;
for(auto e : x) o << ", "+!i++<<e;
return o <<"}";
}
#define debug(x...) cerr << "["#x"]: ",[](auto...$){((cerr<<$<<"; "),...)<<endl;}(x)
#else
#define debug(...) {}
#endif
void solve() {
int N;
cin >> N;
vector<int> a(N);
REP(i, N) cin >> a[i];
vector<int> S;
for (int i = 0; i < N; i++) {
int cur_sum = 0;
for (int j = i; j < N; j++) {
cur_sum += a[j];
S.pb(cur_sum);
}
}
ll cnt = 0;
N = S.size();
sort(all(S));
map<int, int> counter;
for (int i = 0; i < N - 1; i++) {
int current = S[i];
for (int j = i + 1; j < N; j++) {
cnt += (ll)counter[-(current + S[j])];
counter[S[j]] += 1;
}
counter.clear();
}
cout << cnt << '\n';
}
signed main() {
cin.tie(0)->sync_with_stdio(0);
int t = 1;
// cin >> t;
while(t--) {
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 | #include <bits/stdc++.h> using namespace std; #define FOR(i, l, r) for(int i = (l); i <= (r); ++i) #define REP(i, n) FOR(i, 0, (n) - 1) #define rep(i, a, b) for(int i = a; i < (b); ++i) #define all(x) x.begin(), x.end() #define sz(x) (int)(x).size() #define ssize(x) int(x.size()) #define pb push_back #define fi first #define se second #define ld long double typedef long long ll; typedef vector<int> vi; typedef pair<int, int> pii; typedef pair <ld, ld> pd; using LL=long long; #ifdef DEBUG auto &operator<<(auto &o, pair<auto, auto> p) { return o << "()" << p.first << ", " << p.second <<")"; } auto operator<<(auto &o, auto x)-> decltype(x.end(), o) { o << "{";int i = 0; for(auto e : x) o << ", "+!i++<<e; return o <<"}"; } #define debug(x...) cerr << "["#x"]: ",[](auto...$){((cerr<<$<<"; "),...)<<endl;}(x) #else #define debug(...) {} #endif void solve() { int N; cin >> N; vector<int> a(N); REP(i, N) cin >> a[i]; vector<int> S; for (int i = 0; i < N; i++) { int cur_sum = 0; for (int j = i; j < N; j++) { cur_sum += a[j]; S.pb(cur_sum); } } ll cnt = 0; N = S.size(); sort(all(S)); map<int, int> counter; for (int i = 0; i < N - 1; i++) { int current = S[i]; for (int j = i + 1; j < N; j++) { cnt += (ll)counter[-(current + S[j])]; counter[S[j]] += 1; } counter.clear(); } cout << cnt << '\n'; } signed main() { cin.tie(0)->sync_with_stdio(0); int t = 1; // cin >> t; while(t--) { solve(); } } |
English