#include<bits/stdc++.h>
using namespace std;
#define ii pair <int, int>
#define app push_back
mt19937 rnd(228);
struct Node {
int x, y, cnt;
Node *l, *r;
Node (int x_) {
x = x_; y = rnd();
l = r = NULL;
cnt = 1;
}
};
int cnt(Node *t) {
if (!t) return 0;
else return t->cnt;
}
void relax(Node *t) {
t->cnt = cnt(t->l) + cnt(t->r) + 1;
}
Node *merge(Node *l, Node *r) {
if (!l) return r;
if (!r) return l;
if (l->y < r->y) {
l->r = merge(l->r, r);
relax(l);
return l;
}
else {
r->l = merge(l, r->l);
relax(r);
return r;
}
}
pair <Node*, Node*> split(Node *t, int x) {
if (!t) return {NULL, NULL};
if (t->x < x) {
auto tmp = split(t->r, x);
t->r = tmp.first;
relax(t);
return {t, tmp.second};
}
else {
auto tmp = split(t->l, x);
t->l = tmp.second;
relax(t);
return {tmp.first, t};
}
}
void insert(Node *&t, int x) {
auto tmp = split(t, x);
t = merge(merge(tmp.first, new Node(x)), tmp.second);
}
void erase(Node *&t, int x) {
auto tmp = split(t, x);
auto tmp1 = split(tmp.second, x + 1);
t = merge(tmp.first, tmp1.second);
}
Node *get(Node *&t, int l, int r) {
auto tmp = split(t, l);
auto tmp1 = split(tmp.second, r + 1);
t = merge(tmp.first, tmp1.second);
return tmp1.first;
}
const int N = 1e5 + 7;
Node *dd[N][2];
vector <int> g[N];
int vert[N];
int l[N], r[N], ptr = 0;
void dfs(int u, int p) {
l[u] = ++ptr;
vert[ptr] = u;
for (int v : g[u]) {
if (v != p) {
dfs(v, u);
}
}
r[u] = ptr;
}
int mx[N << 2];
void upd(int v, int tl, int tr, int p, int x) {
if (tl == tr) {
mx[v] = x;
return;
}
int tm = (tl + tr) >> 1;
if (p <= tm) upd(v * 2 + 1, tl, tm, p, x);
else upd(v * 2 + 2, tm + 1, tr, p, x);
mx[v] = max(mx[v * 2 + 1], mx[v * 2 + 2]);
}
int parent(int v, int tl, int tr, int l, int r, int x) {
if (tr < l || r < tl || mx[v] < x) return -1;
if (tl == tr) return vert[tl];
int tm = (tl + tr) >> 1;
int t = parent(v * 2 + 2, tm + 1, tr, l, r, x);
if (t != -1) return t;
else return parent(v * 2 + 1, tl, tm, l, r, x);
}
int ans = 0;
void initialize(int n, int *a, int *b) {
for (int i = 0; i < n - 1; ++i) {
g[a[i]].app(b[i]);
g[b[i]].app(a[i]);
}
++n;
g[0].app(1);
dfs(0, 0);
for (int i = 0; i < (N << 2); ++i) {
mx[i] = -1;
}
upd(0, 0, N, l[0], r[0]);
}
//0 - cat, 1 - dog
int type[N];
void add(int u) {
int p = parent(0, 0, N, 0, l[u] - 1, r[u]);
ans += type[u] ^ type[p];
for (int i = 0; i < 2; ++i) {
dd[u][i] = get(dd[p][i], l[u], r[u]);
if (type[u] ^ type[p]) {
if (i == type[u]) {
ans -= cnt(dd[u][i]);
}
else {
ans += cnt(dd[u][i]);
}
}
}
insert(dd[p][type[u]], l[u]);
}
void del(int u) {
int p = parent(0, 0, N, 0, l[u] - 1, r[u]);
ans -= type[u] ^ type[p];
erase(dd[p][type[u]], l[u]);
for (int i = 0; i < 2; ++i) {
if (type[u] ^ type[p]) {
if (i == type[u]) {
ans += cnt(dd[u][i]);
}
else {
ans -= cnt(dd[u][i]);
}
}
auto tmp = split(dd[p][i], l[u]);
auto tmp1 = split(dd[p][i], r[u] + 1);
dd[p][i] = merge(merge(tmp.first, dd[u][i]), tmp1.second);
dd[u][i] = NULL;
}
}
int f() {
if (cnt(dd[0][0]) < cnt(dd[0][1])) return ans;
else return ans - cnt(dd[0][0]) + cnt(dd[0][1]);
}
int cat(int u) {
--u;
type[u] = 0;
add(u);
return f();
}
int dog(int u) {
--u;
type[u] = 1;
add(u);
return f();
}
int neighbor(int u) {
--u;
del(u);
return f();
}
Compilation message
/tmp/ccIqARp8.o: In function `main':
grader.cpp:(.text.startup+0x157): undefined reference to `initialize(int, std::vector<int, std::allocator<int> >, std::vector<int, std::allocator<int> >)'
collect2: error: ld returned 1 exit status
