#include <iostream> constexpr int MAX_N = 500; constexpr int MAX_ABS = 20000; constexpr int ZERO_VAL = MAX_N*MAX_ABS; int sums[ZERO_VAL<<1|1], pref_sum[MAX_N+1]; int main() { std::ios_base::sync_with_stdio(false); std::cin.tie(nullptr); int n; std::cin >> n; for(int i = 0, a; i < n; ++i) { std::cin >> a; pref_sum[i+1] = pref_sum[i]+a; } for(int i = 1; i <= n; ++i) for(int j = 0; j < i; ++j) ++sums[ZERO_VAL-pref_sum[i]+pref_sum[j]]; long long result = 0; for(int i = 2; i <= n; ++i) for(int j = 0; j < i; ++j) { // for(int a = j+2; a < i; ++a) for(int b = j+1; b < a; ++b) // j < b < a < i { --sums[ZERO_VAL-pref_sum[i]+pref_sum[j]]; --sums[ZERO_VAL-pref_sum[a]+pref_sum[b]]; result += sums[ZERO_VAL+pref_sum[i]-pref_sum[j]+pref_sum[a]-pref_sum[b]]; ++sums[ZERO_VAL-pref_sum[i]+pref_sum[j]]; ++sums[ZERO_VAL-pref_sum[a]+pref_sum[b]]; --sums[ZERO_VAL-pref_sum[i]+pref_sum[b]]; --sums[ZERO_VAL-pref_sum[a]+pref_sum[j]]; result += sums[ZERO_VAL+pref_sum[i]-pref_sum[j]+pref_sum[a]-pref_sum[b]]; ++sums[ZERO_VAL-pref_sum[i]+pref_sum[b]]; ++sums[ZERO_VAL-pref_sum[a]+pref_sum[j]]; --sums[ZERO_VAL-pref_sum[i]+pref_sum[a]]; --sums[ZERO_VAL-pref_sum[b]+pref_sum[j]]; result += sums[ZERO_VAL+pref_sum[i]-pref_sum[a]+pref_sum[b]-pref_sum[j]]; ++sums[ZERO_VAL-pref_sum[i]+pref_sum[a]]; ++sums[ZERO_VAL-pref_sum[b]+pref_sum[j]]; } for(int b = j+1; b < i; ++b) // j < b < a = i { --sums[ZERO_VAL-pref_sum[i]+pref_sum[b]]; --sums[ZERO_VAL-pref_sum[i]+pref_sum[j]]; result += sums[ZERO_VAL+pref_sum[i]-pref_sum[j]+pref_sum[i]-pref_sum[b]]; ++sums[ZERO_VAL-pref_sum[i]+pref_sum[b]]; ++sums[ZERO_VAL-pref_sum[i]+pref_sum[j]]; } for(int a = j+1; a < i; ++a) // j = b < a < i { --sums[ZERO_VAL-pref_sum[i]+pref_sum[j]]; --sums[ZERO_VAL-pref_sum[a]+pref_sum[j]]; result += sums[ZERO_VAL+pref_sum[i]-pref_sum[j]+pref_sum[a]-pref_sum[j]]; ++sums[ZERO_VAL-pref_sum[i]+pref_sum[j]]; ++sums[ZERO_VAL-pref_sum[a]+pref_sum[j]]; } for(int a = j+1; a < i; ++a) // j < b = a < i { --sums[ZERO_VAL-pref_sum[i]+pref_sum[a]]; --sums[ZERO_VAL-pref_sum[a]+pref_sum[j]]; result += sums[ZERO_VAL+pref_sum[i]-pref_sum[j]]; ++sums[ZERO_VAL-pref_sum[i]+pref_sum[a]]; ++sums[ZERO_VAL-pref_sum[a]+pref_sum[j]]; } } std::cout << result/3 << '\n'; 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 | #include <iostream> constexpr int MAX_N = 500; constexpr int MAX_ABS = 20000; constexpr int ZERO_VAL = MAX_N*MAX_ABS; int sums[ZERO_VAL<<1|1], pref_sum[MAX_N+1]; int main() { std::ios_base::sync_with_stdio(false); std::cin.tie(nullptr); int n; std::cin >> n; for(int i = 0, a; i < n; ++i) { std::cin >> a; pref_sum[i+1] = pref_sum[i]+a; } for(int i = 1; i <= n; ++i) for(int j = 0; j < i; ++j) ++sums[ZERO_VAL-pref_sum[i]+pref_sum[j]]; long long result = 0; for(int i = 2; i <= n; ++i) for(int j = 0; j < i; ++j) { // for(int a = j+2; a < i; ++a) for(int b = j+1; b < a; ++b) // j < b < a < i { --sums[ZERO_VAL-pref_sum[i]+pref_sum[j]]; --sums[ZERO_VAL-pref_sum[a]+pref_sum[b]]; result += sums[ZERO_VAL+pref_sum[i]-pref_sum[j]+pref_sum[a]-pref_sum[b]]; ++sums[ZERO_VAL-pref_sum[i]+pref_sum[j]]; ++sums[ZERO_VAL-pref_sum[a]+pref_sum[b]]; --sums[ZERO_VAL-pref_sum[i]+pref_sum[b]]; --sums[ZERO_VAL-pref_sum[a]+pref_sum[j]]; result += sums[ZERO_VAL+pref_sum[i]-pref_sum[j]+pref_sum[a]-pref_sum[b]]; ++sums[ZERO_VAL-pref_sum[i]+pref_sum[b]]; ++sums[ZERO_VAL-pref_sum[a]+pref_sum[j]]; --sums[ZERO_VAL-pref_sum[i]+pref_sum[a]]; --sums[ZERO_VAL-pref_sum[b]+pref_sum[j]]; result += sums[ZERO_VAL+pref_sum[i]-pref_sum[a]+pref_sum[b]-pref_sum[j]]; ++sums[ZERO_VAL-pref_sum[i]+pref_sum[a]]; ++sums[ZERO_VAL-pref_sum[b]+pref_sum[j]]; } for(int b = j+1; b < i; ++b) // j < b < a = i { --sums[ZERO_VAL-pref_sum[i]+pref_sum[b]]; --sums[ZERO_VAL-pref_sum[i]+pref_sum[j]]; result += sums[ZERO_VAL+pref_sum[i]-pref_sum[j]+pref_sum[i]-pref_sum[b]]; ++sums[ZERO_VAL-pref_sum[i]+pref_sum[b]]; ++sums[ZERO_VAL-pref_sum[i]+pref_sum[j]]; } for(int a = j+1; a < i; ++a) // j = b < a < i { --sums[ZERO_VAL-pref_sum[i]+pref_sum[j]]; --sums[ZERO_VAL-pref_sum[a]+pref_sum[j]]; result += sums[ZERO_VAL+pref_sum[i]-pref_sum[j]+pref_sum[a]-pref_sum[j]]; ++sums[ZERO_VAL-pref_sum[i]+pref_sum[j]]; ++sums[ZERO_VAL-pref_sum[a]+pref_sum[j]]; } for(int a = j+1; a < i; ++a) // j < b = a < i { --sums[ZERO_VAL-pref_sum[i]+pref_sum[a]]; --sums[ZERO_VAL-pref_sum[a]+pref_sum[j]]; result += sums[ZERO_VAL+pref_sum[i]-pref_sum[j]]; ++sums[ZERO_VAL-pref_sum[i]+pref_sum[a]]; ++sums[ZERO_VAL-pref_sum[a]+pref_sum[j]]; } } std::cout << result/3 << '\n'; return 0; } |