#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define int ll
#define x first
#define y second
#define mp make_pair
#define mt make_tuple
#define all(x) x.begin(), x.end()
const int NODES = 6e5 + 10;
mt19937 rnd;
struct node {
int x, w, dy, sy, ls, rs;
} pool[NODES];
int ff = 1;
#define getter(f) int &f(int v) { assert(0 <= v && v < NODES); return pool[v].f; }
getter(x) getter(w) getter(dy) getter(sy) getter(ls) getter(rs)
int new_node(int nx, int ny) {
assert(ff < NODES);
int v = ff++;
x(v) = nx;
w(v) = (int)rnd();
dy(v) = ny;
sy(v) = ny;
ls(v) = rs(v) = 0;
return v;
}
int pull(int v) {
sy(v) = sy(ls(v)) + dy(v) + sy(rs(v));
return v;
}
void to_vector(int t, vector<pair<int, int>> &vec) {
if (!t) {
return;
}
to_vector(ls(t), vec);
vec.emplace_back(x(t), dy(t));
to_vector(rs(t), vec);
}
void print(int t) {
vector<pair<int, int>> v;
to_vector(t, v);
for (auto e : v) {
cout << "(" << e.x << "; " << e.y << ") ";
}
cout << endl;
}
pair<int, int> splitsum(int t, int s) {
if (!t) {
return {0, 0};
}
assert(0 <= s && s <= sy(t));
int lsum = sy(ls(t)), msum = dy(t);
if (lsum < s && lsum + msum > s) {
int lt = new_node(x(t), s - lsum), rt = new_node(x(t), msum - dy(lt));
ls(lt) = ls(t), rs(rt) = rs(t);
return {lt, rt};
}
int l, r;
if (s <= lsum) {
tie(l, ls(t)) = splitsum(ls(t), s);
r = pull(t);
} else {
tie(rs(t), r) = splitsum(rs(t), s - lsum - msum);
l = pull(t);
}
assert(sy(l) == s);
return {l, r};
}
// (-inf, k], (k, +inf)
pair<int, int> split(int t, int k) {
if (!t) {
return {0, 0};
}
int l, r;
if (k < x(t)) {
tie(l, ls(t)) = split(ls(t), k);
r = pull(t);
} else {
tie(rs(t), r) = split(rs(t), k);
l = pull(t);
}
return {l, r};
}
int merge(int l, int r) {
if (!l) {
return r;
}
if (!r) {
return l;
}
if (w(l) < w(r)) {
rs(l) = merge(rs(l), r);
return pull(l);
} else {
ls(r) = merge(l, ls(r));
return pull(r);
}
}
int get_value(int t, int x) {
int l, r;
tie(l, r) = split(t, x);
int res = sy(l);
t = merge(l, r);
return res;
}
__attribute__((warn_unused_result))
int insert(int t, int nx, int ny) {
int l, m, r;
tie(l, m) = split(t, nx - 1);
tie(m, r) = split(m, nx);
// cout << "insert " << nx << " " << ny << endl;
// print(l);
// print(m);
// print(r);
// cout << "---" << endl;
m = new_node(nx, sy(m) + ny);
l = merge(l, m);
// print(l);
return merge(l, r);
}
__attribute__((warn_unused_result))
int uminpref(int t, int px) {
int checksum = sy(t);
int l, m, r;
tie(l, m) = split(t, px - 1);
tie(m, r) = split(m, px);
if (sy(m) <= 0) {
int p, q;
tie(p, q) = splitsum(l, sy(l) + sy(m));
t = merge(p, r);
assert(sy(t) == checksum);
return t;
} else {
return merge(merge(l, m), r);
}
}
__attribute__((warn_unused_result))
int add(int tv, int tu) {
if (!tv) {
return tu;
}
if (!tu) {
return tv;
}
tv = insert(tv, x(tu), dy(tu));
tv = add(tv, ls(tu));
tv = add(tv, rs(tu));
return tv;
}
const int N = 2e5 + 10;
int n, ia[N], ih[N], ic[N], dp[N];
vector<int> g[N];
void dfs(int v) {
for (int u : g[v]) {
dfs(u);
dp[u] = uminpref(dp[u], ih[u]);
// cout << "v = " << u + 1 << endl;
// print(dp[u]);
dp[v] = add(dp[v], dp[u]);
}
dp[v] = insert(dp[v], 0, ic[v]);
dp[v] = insert(dp[v], ih[v], -ic[v]);
dp[v] = insert(dp[v], ih[v] + 1, ic[v]);
}
signed main() {
#ifdef LC
assert(freopen("input.txt", "r", stdin));
#endif
ios::sync_with_stdio(0); cin.tie(0);
cin >> n;
for (int i = 0; i < n; ++i) {
cin >> ia[i] >> ih[i] >> ic[i];
if (--ia[i] == i) {
continue;
}
g[ia[i]].push_back(i);
}
dfs(0);
// print(dp[0]);
dp[0] = uminpref(dp[0], ih[0]);
cout << get_value(dp[0], 0) << "\n";
return 0;
}
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
3 ms |
5028 KB |
Output is correct |
| 2 |
Correct |
3 ms |
5024 KB |
Output is correct |
| 3 |
Correct |
4 ms |
4940 KB |
Output is correct |
| 4 |
Correct |
4 ms |
4940 KB |
Output is correct |
| 5 |
Runtime error |
12 ms |
10600 KB |
Execution killed with signal 6 |
| 6 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Runtime error |
12 ms |
10672 KB |
Execution killed with signal 6 |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
3 ms |
5028 KB |
Output is correct |
| 2 |
Correct |
3 ms |
5024 KB |
Output is correct |
| 3 |
Correct |
4 ms |
4940 KB |
Output is correct |
| 4 |
Correct |
4 ms |
4940 KB |
Output is correct |
| 5 |
Runtime error |
12 ms |
10600 KB |
Execution killed with signal 6 |
| 6 |
Halted |
0 ms |
0 KB |
- |
