#include <algorithm>
#include <cstdio>
#include <map>
using namespace std;
#define FOR(i,a,b) for(int i=(a);i<(b);++i)
#define REP(i,n) FOR(i,0,n)
#define VAR(v,w) __typeof(w) v=(w)
#define FORE(it,c) for(VAR(it,(c).begin());it!=(c).end();++it)
#define FT first
#define SD second
#define INT(x) int x; scanf("%d", &x)
typedef long long LL;
int a[500], s[501];
int main() {
INT(n);
REP(i,n) scanf("%d", &a[i]);
s[0] = 0;
REP(i,n) s[i + 1] = s[i] + a[i];
map<int, int> m;
REP(i,n) FOR(j,i+1,n+1) ++m[s[j] - s[i]];
LL r = 0;
FORE(it,m) FORE(jt,m) {
int x = -it->FT - jt->FT;
VAR(kt, m.find(x));
if (kt == m.end()) continue;
LL a = it->SD;
LL b = max(jt->SD - (it->FT == jt->FT), 0);
LL c = max(kt->SD - (it->FT == kt->FT) - (jt->FT == kt->FT), 0);
r += a * b * c;
}
r /= 6;
printf("%lld\n", r);
}
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 | #include <algorithm> #include <cstdio> #include <map> using namespace std; #define FOR(i,a,b) for(int i=(a);i<(b);++i) #define REP(i,n) FOR(i,0,n) #define VAR(v,w) __typeof(w) v=(w) #define FORE(it,c) for(VAR(it,(c).begin());it!=(c).end();++it) #define FT first #define SD second #define INT(x) int x; scanf("%d", &x) typedef long long LL; int a[500], s[501]; int main() { INT(n); REP(i,n) scanf("%d", &a[i]); s[0] = 0; REP(i,n) s[i + 1] = s[i] + a[i]; map<int, int> m; REP(i,n) FOR(j,i+1,n+1) ++m[s[j] - s[i]]; LL r = 0; FORE(it,m) FORE(jt,m) { int x = -it->FT - jt->FT; VAR(kt, m.find(x)); if (kt == m.end()) continue; LL a = it->SD; LL b = max(jt->SD - (it->FT == jt->FT), 0); LL c = max(kt->SD - (it->FT == kt->FT) - (jt->FT == kt->FT), 0); r += a * b * c; } r /= 6; printf("%lld\n", r); } |
English