#include<bits/stdc++.h> using namespace std; long long tab[509], val[60000009]; const long long M = 30000000; int main() { ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0); long long n, a, wyn = 0, tro = 0; cin>>n; for(int i=1; i<=n; i++) { cin>>tab[i]; tab[i] += tab[i-1]; } for(int i=1; i<=n; i++) { for(int j=1; j<i; j++) { for(int k=0; k<j; k++) { a = - tab[i-1] + tab[j] - tab[k]; val[a+M]++; } } for(int j=1; j<=i; j++) { for(int k=0; k<j; k++) { a = tab[i] + tab[j] - tab[k]; wyn += val[-a+M]; } } for(int x=0; x<=i-1; x++) { for(int k=0; k<=i-1; k++) { a = - tab[x] + tab[i] - tab[k]; val[a+M]++; } } } for(int i=1; i<=n; i++) { for(int j=1; j<i; j++) { for(int k=0; k<j; k++) { a = - tab[i-1] + tab[j] - tab[k]; val[a+M]--; } } for(int x=0; x<=i-1; x++) { for(int k=0; k<=i-1; k++) { a = - tab[x] + tab[i] - tab[k]; val[a+M]--; } } } for(int i=1; i<=n; i++) { for(int j=0; j<i; j++) { for(int k=0; k<j; k++) { a = 3 * tab[i] - tab[j] - tab[k]; if(abs(a) <= 10000000) tro += val[-a+M]; val[-tab[k]+M]++; } for(int k=0; k<j; k++) val[-tab[k]+M]--; } } for(int i=1; i<=n; i++) for(int j=0; j<i; j++) val[tab[i]-tab[j]+M]++; for(int i=1; i<=n; i++) { for(int j=0; j<i; j++) val[tab[i]-tab[j]+M]--; for(int j=0; j<i; j++) { a = 2 * (tab[i] - tab[j]); if(abs(a) <= 20000000) wyn -= val[-a+M]; } for(int j=0; j<i; j++) val[tab[i]-tab[j]+M]++; } wyn /= 2; cout<<wyn+tro; 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 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 | #include<bits/stdc++.h> using namespace std; long long tab[509], val[60000009]; const long long M = 30000000; int main() { ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0); long long n, a, wyn = 0, tro = 0; cin>>n; for(int i=1; i<=n; i++) { cin>>tab[i]; tab[i] += tab[i-1]; } for(int i=1; i<=n; i++) { for(int j=1; j<i; j++) { for(int k=0; k<j; k++) { a = - tab[i-1] + tab[j] - tab[k]; val[a+M]++; } } for(int j=1; j<=i; j++) { for(int k=0; k<j; k++) { a = tab[i] + tab[j] - tab[k]; wyn += val[-a+M]; } } for(int x=0; x<=i-1; x++) { for(int k=0; k<=i-1; k++) { a = - tab[x] + tab[i] - tab[k]; val[a+M]++; } } } for(int i=1; i<=n; i++) { for(int j=1; j<i; j++) { for(int k=0; k<j; k++) { a = - tab[i-1] + tab[j] - tab[k]; val[a+M]--; } } for(int x=0; x<=i-1; x++) { for(int k=0; k<=i-1; k++) { a = - tab[x] + tab[i] - tab[k]; val[a+M]--; } } } for(int i=1; i<=n; i++) { for(int j=0; j<i; j++) { for(int k=0; k<j; k++) { a = 3 * tab[i] - tab[j] - tab[k]; if(abs(a) <= 10000000) tro += val[-a+M]; val[-tab[k]+M]++; } for(int k=0; k<j; k++) val[-tab[k]+M]--; } } for(int i=1; i<=n; i++) for(int j=0; j<i; j++) val[tab[i]-tab[j]+M]++; for(int i=1; i<=n; i++) { for(int j=0; j<i; j++) val[tab[i]-tab[j]+M]--; for(int j=0; j<i; j++) { a = 2 * (tab[i] - tab[j]); if(abs(a) <= 20000000) wyn -= val[-a+M]; } for(int j=0; j<i; j++) val[tab[i]-tab[j]+M]++; } wyn /= 2; cout<<wyn+tro; return 0; } |