#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using pl = pair<ll, ll>;
using pi = pair<int, int>;
#define vt vector
#define f first
#define s second
#define pb push_back
#define all(x) x.begin(), x.end()
#define size(x) ((int) (x).size())
#define FOR(i, a, b) for (int i = (a); i < (b); i++)
#define ROF(i, a, b) for (int i = (b) - 1; i >= (a); i--)
#define F0R(i, b) FOR (i, 0, b)
#define endl '\n'
const ll INF = 1e18;
const int inf = 1e9;
template<template<typename> class Container, typename T>
ostream& operator<<(ostream& os, Container<T> o) {
os << "{";
int g = size(o);
for (auto i : o) os << i << ((--g) == 0 ? "" : ", ");
os << "}";
return os;
}
void _print() {
cerr << "\n";
}
template<typename T, typename ...V>
void _print(T t, V... v) {
cerr << t; if (sizeof...(v)) cerr << ", "; _print(v...);
}
#ifdef LOCAL
#define dbg(x...) cerr << #x << " = "; _print(x);
#else
#define dbg(x...)
#define cerr if (0) std::cerr
#endif
/*
removing a ball just deletes its most ancestor ancestor
balls have a fixed "preferred path" at any given state
if a node has a ball in it, then so does its entire subtree
given some tree, there is a fixed priority among active nodes
*/
#ifdef LOCAL
const int sz = 1 << 4;
#else
const int sz = 1 << 17;
#endif
struct LazySeg {
vt<int> seg, lazy;
void init() {
seg.resize(2 * sz, 1);
lazy.resize(2 * sz, -1);
ROF (i, 1, sz) seg[i] = seg[2 * i] + seg[2 * i + 1];
}
void pull(int i) {
seg[i] = seg[2 * i] + seg[2 * i + 1];
}
void push(int i, int l, int r) {
if (lazy[i] == -1) return;
seg[i] = lazy[i] * (r - l + 1);
if (l != r) F0R (j, 2) lazy[2 * i + j] = lazy[i];
lazy[i] = -1;
}
void upd(int lo, int hi, int v, int i = 1, int l = 0, int r = sz - 1) {
push(i, l, r);
if (lo > r || hi < l) return;
if (lo <= l && r <= hi) {
lazy[i] = v;
push(i, l, r);
return;
}
int m = (l + r) / 2;
upd(lo, hi, v, 2 * i, l, m);
upd(lo, hi, v, 2 * i + 1, m + 1, r);
pull(i);
}
int query(int lo, int hi, int i = 1, int l = 0, int r = sz - 1) {
push(i, l, r);
if (lo > r || hi < l) return 0;
if (lo <= l && r <= hi) return seg[i];
int m = (l + r) / 2;
return query(lo, hi, 2 * i, l, m) + query(lo, hi, 2 * i + 1, m + 1, r);
}
int walk(int k, int i = 1, int l = 0, int r = sz - 1) {
if (l == r) return l;
int m = (l + r) / 2;
push(2 * i, l, m), push(2 * i + 1, m + 1, r);
if (seg[2 * i] >= k) return walk(k, 2 * i, l, m);
else return walk(k - seg[2 * i], 2 * i + 1, m + 1, r);
}
};
int n, q, t;
vt<int> adj[100005];
vt<int> depth, pri, rpri, jmp, par, mn;
LazySeg seg;
int dp(int u) {
int res = u;
for (int v : adj[u]) {
res = min(res, dp(v));
}
return mn[u] = res;
}
void dfs(int u, int p = -1) {
par[u] = p;
for (int v : adj[u]) {
depth[v] = depth[u] + 1;
if (depth[u] + depth[jmp[jmp[u]]] == 2 * depth[jmp[u]]) jmp[v] = jmp[jmp[u]];
else jmp[v] = u;
dfs(v, u);
}
rpri[t] = u;
pri[u] = t++;
}
int lift(int u) {
auto good = [&] (int v) {
return seg.query(pri[v], pri[v]) == 0;
};
while (par[u] != u && good(par[u])) {
if (good(jmp[u])) u = jmp[u];
else u = par[u];
}
return u;
}
main() {
cin.tie(0)->sync_with_stdio(0);
cin >> n >> q;
depth = pri = rpri = jmp = par = mn = vt<int>(n);
int rt = -1;
F0R (i, n) {
int p; cin >> p;
p--;
if (p >= 0) {
par[i] = p;
adj[p].pb(i);
} else {
rt = i;
}
}
seg.init();
dp(rt);
F0R (i, n) sort(all(adj[i]), [&] (int u, int v) { return mn[u] < mn[v]; });
dfs(rt, rt);
F0R (i, q) {
int t, u; cin >> t >> u;
if (t == 1) { // insert balls
int r = seg.walk(u);
seg.upd(0, r, 0);
cout << rpri[r] + 1 << endl;
} else if (t == 2) {
u--;
int v = lift(u);
seg.upd(pri[v], pri[v], 1);
cout << depth[u] - depth[v] << endl;
} else assert(0);
}
}
Compilation message
ballmachine.cpp:145:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
145 | main() {
| ^~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Execution timed out |
1064 ms |
4700 KB |
Time limit exceeded |
| 2 |
Execution timed out |
1067 ms |
7260 KB |
Time limit exceeded |
| 3 |
Correct |
42 ms |
7760 KB |
Output is correct |
| 4 |
Correct |
2 ms |
4952 KB |
Output is correct |
| 5 |
Execution timed out |
1100 ms |
4700 KB |
Time limit exceeded |
| 6 |
Execution timed out |
1012 ms |
4696 KB |
Time limit exceeded |
| 7 |
Execution timed out |
1012 ms |
4696 KB |
Time limit exceeded |
| 8 |
Execution timed out |
1018 ms |
4696 KB |
Time limit exceeded |
| 9 |
Execution timed out |
1055 ms |
4956 KB |
Time limit exceeded |
| 10 |
Execution timed out |
1036 ms |
5468 KB |
Time limit exceeded |
| 11 |
Execution timed out |
1041 ms |
7260 KB |
Time limit exceeded |
| 12 |
Correct |
39 ms |
7760 KB |
Output is correct |
| 13 |
Execution timed out |
1018 ms |
7260 KB |
Time limit exceeded |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
37 ms |
7252 KB |
Output is correct |
| 2 |
Correct |
61 ms |
12368 KB |
Output is correct |
| 3 |
Correct |
49 ms |
8656 KB |
Output is correct |
| 4 |
Correct |
37 ms |
7504 KB |
Output is correct |
| 5 |
Correct |
35 ms |
7512 KB |
Output is correct |
| 6 |
Correct |
47 ms |
7360 KB |
Output is correct |
| 7 |
Correct |
40 ms |
6992 KB |
Output is correct |
| 8 |
Correct |
29 ms |
7760 KB |
Output is correct |
| 9 |
Correct |
63 ms |
13652 KB |
Output is correct |
| 10 |
Correct |
84 ms |
13400 KB |
Output is correct |
| 11 |
Correct |
62 ms |
13328 KB |
Output is correct |
| 12 |
Correct |
73 ms |
11780 KB |
Output is correct |
| 13 |
Correct |
57 ms |
16380 KB |
Output is correct |
| 14 |
Correct |
50 ms |
9468 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Execution timed out |
1020 ms |
9048 KB |
Time limit exceeded |
| 2 |
Execution timed out |
1079 ms |
11128 KB |
Time limit exceeded |
| 3 |
Execution timed out |
1050 ms |
14684 KB |
Time limit exceeded |
| 4 |
Execution timed out |
1069 ms |
11504 KB |
Time limit exceeded |
| 5 |
Correct |
146 ms |
11780 KB |
Output is correct |
| 6 |
Correct |
134 ms |
11604 KB |
Output is correct |
| 7 |
Execution timed out |
1039 ms |
10064 KB |
Time limit exceeded |
| 8 |
Execution timed out |
1060 ms |
14564 KB |
Time limit exceeded |
| 9 |
Execution timed out |
1047 ms |
13392 KB |
Time limit exceeded |
| 10 |
Execution timed out |
1057 ms |
12636 KB |
Time limit exceeded |
| 11 |
Execution timed out |
1103 ms |
12620 KB |
Time limit exceeded |
| 12 |
Execution timed out |
1070 ms |
11100 KB |
Time limit exceeded |
| 13 |
Execution timed out |
1042 ms |
16976 KB |
Time limit exceeded |
| 14 |
Correct |
81 ms |
9676 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Execution timed out |
1053 ms |
13148 KB |
Time limit exceeded |
| 2 |
Execution timed out |
1039 ms |
11148 KB |
Time limit exceeded |
| 3 |
Correct |
61 ms |
17648 KB |
Output is correct |
| 4 |
Execution timed out |
1058 ms |
13148 KB |
Time limit exceeded |
| 5 |
Correct |
260 ms |
13556 KB |
Output is correct |
| 6 |
Correct |
184 ms |
13396 KB |
Output is correct |
| 7 |
Execution timed out |
1072 ms |
11100 KB |
Time limit exceeded |
| 8 |
Correct |
59 ms |
17744 KB |
Output is correct |
| 9 |
Correct |
44 ms |
9420 KB |
Output is correct |
