// // Mateusz Pietrowcow // #include <bits/stdc++.h> #include <ext/pb_ds/assoc_container.hpp> #define MOD 1000000007 #define INF 1000000000 #define INFL 1000000000000000000LL #define Tak cout << "TAK" << '\n' #define Nie cout << "NIE" << '\n' #define min3(x, y, z) min(x, min(y, z)) #define max3(x, y, z) max(x, max(y, z)) #define all(x) x.begin(),x.end() using namespace std; typedef long long ll; typedef unsigned int uint; typedef unsigned long long ull; typedef pair<int,int> ii; typedef pair<long long, long long> pll; typedef pair<unsigned long long, unsigned long long> pull; constexpr ll N = 5e2 + 5, MAX = 6e7 + 5, add = 3e7; ll t[N], r[N*N], c[MAX]; ll n, m = 0, sum; void read() { cin >> n; for (int i = 0; i < n; i++) cin >> t[i]; } void solve() { ull res = 0; for (int i = 0; i < n; i++) { sum = 0; for (int j = i; j < n; j++) { sum += t[j]; r[m++] = sum; c[sum + add]++; } } sum = 0; for (int i = 0; i < m; i++) { sum -= r[i]; for (int j = i + 1; j < m; j++) { sum -= r[j]; res += c[sum + add] - (sum == r[j]) - (sum == r[i]); sum += r[j]; } sum += r[i]; } cout << res / 3LL << '\n'; } int main() { ios_base::sync_with_stdio(0); cout.tie(0); cin.tie(0); read(); solve(); }
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 | // // Mateusz Pietrowcow // #include <bits/stdc++.h> #include <ext/pb_ds/assoc_container.hpp> #define MOD 1000000007 #define INF 1000000000 #define INFL 1000000000000000000LL #define Tak cout << "TAK" << '\n' #define Nie cout << "NIE" << '\n' #define min3(x, y, z) min(x, min(y, z)) #define max3(x, y, z) max(x, max(y, z)) #define all(x) x.begin(),x.end() using namespace std; typedef long long ll; typedef unsigned int uint; typedef unsigned long long ull; typedef pair<int,int> ii; typedef pair<long long, long long> pll; typedef pair<unsigned long long, unsigned long long> pull; constexpr ll N = 5e2 + 5, MAX = 6e7 + 5, add = 3e7; ll t[N], r[N*N], c[MAX]; ll n, m = 0, sum; void read() { cin >> n; for (int i = 0; i < n; i++) cin >> t[i]; } void solve() { ull res = 0; for (int i = 0; i < n; i++) { sum = 0; for (int j = i; j < n; j++) { sum += t[j]; r[m++] = sum; c[sum + add]++; } } sum = 0; for (int i = 0; i < m; i++) { sum -= r[i]; for (int j = i + 1; j < m; j++) { sum -= r[j]; res += c[sum + add] - (sum == r[j]) - (sum == r[i]); sum += r[j]; } sum += r[i]; } cout << res / 3LL << '\n'; } int main() { ios_base::sync_with_stdio(0); cout.tie(0); cin.tie(0); read(); solve(); } |