#include <vector>
#include <cstdio>
#include <stack>
#include <algorithm>
#include <queue>
#include <map>
#include <cmath>
#include <unordered_set>
#include <unordered_map>
#include <set>
#include <cstring>
#include <bitset>
#include <iostream>
#include <queue>
#include <iomanip>
#include <complex>
using namespace std;
#define ALL(x) x.begin(), x.end()
#define UNIQUE(c) (c).resize(unique(ALL(c)) - (c).begin())
#define FOR(i, a, b) for(int i =(a); i <=(b); ++i)
#define RE(i, n) FOR(i, 1, n)
#define RED(i, n) FORD(i, n, 1)
#define FORD(i, a, b) for(int i = (a); i >= (b); --i)
#define REP(i, n) for(int i = 0;i <(n); ++i)
#define REPD(i, n) FORD(i, n-1,0)
const int maxn = 510, maxm = 1010;
const long long mod = 998244353, inf = 1000000004000000004ll;
int n,m, a[maxn], mi = 20001, b[maxn * maxn], il, t[20000002];
long long res;
void solve() {
cin>>n;
REP(i, n) {
cin>>a[i];
}
REP(po, n) {
int su = 0;
FOR(ko, po, n - 1){
su += a[ko];
b[il] = su;
mi = min(mi, b[il++]);
}
}
if(mi > 0) {
cout<<"0\n";
return;
}
REP(i, il){
b[i] -= mi;
// cout<<b[i]<<" ";
t[b[i]]++;
}
// cout<<"\n";
REP(i, il){
REP(j, i){
if(b[i] * 2 + b[j] == 3 * (-mi))res--;
if(b[i] + b[j] * 2 == 3 * (-mi))res--;
if(3 * (-mi) - b[i] - b[j] >= 0 && 3 * (-mi) - b[i] - b[j] < 20000002 && t[3 * (-mi) - b[i] - b[j]]) {
res += t[3 * (-mi) - b[i] - b[j]];
// cout<<b[i] + mi<<" "<<b[j] + mi<<" "<<t[3 * (-mi) - b[i] - b[j]]<<" "<<3 * (-mi) - b[i] - b[j]<<"\n";
}
}
}
cout<<res/3<<"\n";
}
int main() {
ios_base::sync_with_stdio(false), cin.tie(nullptr);
int tt = 1;
// cin >> tt;
while (tt--) {
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 | #include <vector> #include <cstdio> #include <stack> #include <algorithm> #include <queue> #include <map> #include <cmath> #include <unordered_set> #include <unordered_map> #include <set> #include <cstring> #include <bitset> #include <iostream> #include <queue> #include <iomanip> #include <complex> using namespace std; #define ALL(x) x.begin(), x.end() #define UNIQUE(c) (c).resize(unique(ALL(c)) - (c).begin()) #define FOR(i, a, b) for(int i =(a); i <=(b); ++i) #define RE(i, n) FOR(i, 1, n) #define RED(i, n) FORD(i, n, 1) #define FORD(i, a, b) for(int i = (a); i >= (b); --i) #define REP(i, n) for(int i = 0;i <(n); ++i) #define REPD(i, n) FORD(i, n-1,0) const int maxn = 510, maxm = 1010; const long long mod = 998244353, inf = 1000000004000000004ll; int n,m, a[maxn], mi = 20001, b[maxn * maxn], il, t[20000002]; long long res; void solve() { cin>>n; REP(i, n) { cin>>a[i]; } REP(po, n) { int su = 0; FOR(ko, po, n - 1){ su += a[ko]; b[il] = su; mi = min(mi, b[il++]); } } if(mi > 0) { cout<<"0\n"; return; } REP(i, il){ b[i] -= mi; // cout<<b[i]<<" "; t[b[i]]++; } // cout<<"\n"; REP(i, il){ REP(j, i){ if(b[i] * 2 + b[j] == 3 * (-mi))res--; if(b[i] + b[j] * 2 == 3 * (-mi))res--; if(3 * (-mi) - b[i] - b[j] >= 0 && 3 * (-mi) - b[i] - b[j] < 20000002 && t[3 * (-mi) - b[i] - b[j]]) { res += t[3 * (-mi) - b[i] - b[j]]; // cout<<b[i] + mi<<" "<<b[j] + mi<<" "<<t[3 * (-mi) - b[i] - b[j]]<<" "<<3 * (-mi) - b[i] - b[j]<<"\n"; } } } cout<<res/3<<"\n"; } int main() { ios_base::sync_with_stdio(false), cin.tie(nullptr); int tt = 1; // cin >> tt; while (tt--) { solve(); } return 0; } |
English