#include <bits/stdc++.h>
#ifdef LOC
#include "debuglib.h"
#else
#define deb(...)
#define DBP(...)
#endif
using namespace std;
using ll = long long;
using vi = vector<int>;
using pii = pair<int, int>;
#define pb push_back
#define mp make_pair
#define x first
#define y second
#define rep(i, b, e) for (int i = (b); i < (e); i++)
#define each(a, x) for (auto& a : (x))
#define all(x) (x).begin(), (x).end()
#define sz(x) int((x).size())
constexpr int MAX_N = 500;
constexpr int MAX_V = 20'000;
constexpr int SUM = MAX_N * MAX_V * 3 + 5;
int bufTriples[SUM*2], bufPairs[SUM*2];
int *triples = bufTriples+SUM, *pairs = bufPairs+SUM;
void run() {
int n; cin >> n;
vi pref(n+1);
rep(i, 0, n) {
cin >> pref[i+1];
pref[i+1] += pref[i];
}
ll ans = 0;
for (int k = n-1; k >= 0; k--) {
rep(i, 0, n+1) rep(j, i+1, n+1) {
int a = pref[k+1], b = pref[i], c = pref[j];
triples[a-b+c]++;
}
rep(i, 0, n+1) rep(j, i+1, n+1) {
int a = pref[k], b = pref[i], c = pref[j];
ans += triples[a+b-c];
}
}
rep(i, 0, n+1) rep(j, i+1, n+1) {
int a = pref[i], b = pref[j];
pairs[b-a]++;
}
rep(i, 0, n+1) rep(j, i+1, n+1) {
int a = pref[i], b = pref[j];
ans -= pairs[(a-b)*2] * 3;
ans += (a == b) * 2;
}
ans /= 6;
cout << ans << '\n';
}
int main() {
cin.sync_with_stdio(0); cin.tie(0);
cout << fixed << setprecision(10);
run();
cout << flush; _Exit(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 | #include <bits/stdc++.h> #ifdef LOC #include "debuglib.h" #else #define deb(...) #define DBP(...) #endif using namespace std; using ll = long long; using vi = vector<int>; using pii = pair<int, int>; #define pb push_back #define mp make_pair #define x first #define y second #define rep(i, b, e) for (int i = (b); i < (e); i++) #define each(a, x) for (auto& a : (x)) #define all(x) (x).begin(), (x).end() #define sz(x) int((x).size()) constexpr int MAX_N = 500; constexpr int MAX_V = 20'000; constexpr int SUM = MAX_N * MAX_V * 3 + 5; int bufTriples[SUM*2], bufPairs[SUM*2]; int *triples = bufTriples+SUM, *pairs = bufPairs+SUM; void run() { int n; cin >> n; vi pref(n+1); rep(i, 0, n) { cin >> pref[i+1]; pref[i+1] += pref[i]; } ll ans = 0; for (int k = n-1; k >= 0; k--) { rep(i, 0, n+1) rep(j, i+1, n+1) { int a = pref[k+1], b = pref[i], c = pref[j]; triples[a-b+c]++; } rep(i, 0, n+1) rep(j, i+1, n+1) { int a = pref[k], b = pref[i], c = pref[j]; ans += triples[a+b-c]; } } rep(i, 0, n+1) rep(j, i+1, n+1) { int a = pref[i], b = pref[j]; pairs[b-a]++; } rep(i, 0, n+1) rep(j, i+1, n+1) { int a = pref[i], b = pref[j]; ans -= pairs[(a-b)*2] * 3; ans += (a == b) * 2; } ans /= 6; cout << ans << '\n'; } int main() { cin.sync_with_stdio(0); cin.tie(0); cout << fixed << setprecision(10); run(); cout << flush; _Exit(0); } |
English