//
// 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(); } |
English