动态点分治模板题。
题意
一棵树,每个点可以是黑色或者白色。
有一些询问,一种是修改一个点的颜色,另一种查询整棵树中距离最远的两个黑点的距离。
思路
还有一种做法是括号序列,详见某岛?,这里不赘述了,因为并不会并不是我写这一题的重点。
所以考虑建出点分树。然后在每个点维护两个堆:
- u作为重心的管辖区域内,所有黑点到u在点分树上的父亲的距离。
- u在点分树上的所有子树的1堆的堆顶。
然后再来一个堆统计答案。
为什么这样做?我们回忆一下没有动态的点分治,在 c a l c ( ) calc() calc()函数里我们干了什么。以当前重心为根遍历了他的管辖区域并且统计答案。那么同理,这里只是用数据结构来维护这个答案而已。
最后要做到在堆中删除需要再开一个辅助堆,然后略微注意下细节。
代码
长的不行。明明别人150行就写完了。
#include<bits/stdc++.h>
using namespace std;
#define rdi read<int>
#define rdl read<long long>
template<typename T> inline T read()
{
T x = 0, fh = 1;
char c = getchar();
while (c < '0' || c > '9'){if (c == '-') fh = -1; c = getchar();}
while (c >= '0' && c <= '9') x = (x<<3)+(x<<1)+c-'0', c = getchar();
return x*fh;
}
template<class T>inline void chkMax(T &x, T y){if (x < y) x = y;}
template<class T>inline void chkMin(T &x, T y){if (x > y) x = y;}
const int inf = 1e9+7;
const int N = 1e5+10;
const int M = N<<1;
const int E = 20;
struct G{
int h[N], e, nxt[M], v[M];
inline void clear(){
e = 1;
}
inline void add_dir(int _u, int _v){
++ e;
nxt[e] = h[_u];
v[e] = _v;
h[_u] = e;
}
inline void add_undir(int _u, int _v){
add_dir(_u, _v);
add_dir(_v, _u);
}
}g;
struct Heap{
priority_queue<int> q, d;
int sz;
Heap(){sz = 0;}
inline void add(int x){q.push(x), ++ sz;}
inline void del(int x){d.push(x); -- sz;}
inline int top(){
while (!d.empty() && q.top() == d.top())
q.pop(), d.pop();
if (sz == 0) return 0;
return q.top();
}
}a[N], b[N], ans;
int n, q, r[N], litn;
int dep[N], dfn[N<<1], idx, fs[N], ed[N], lg[N<<1], minn[N<<1][E];
bool lit[N];
int sz[N], dpt[N], siz, mx, zx;
bool vis[N];
void dfs(int u, int fa)
{
dep[u] = dep[fa]+1;
dfn[fs[u] = ed[u] = ++idx] = u;
for (int i = g.h[u]; i; i = g.nxt[i]){
int v = g.v[i];
if (v == fa) continue;
dfs(v, u);
dfn[ed[u] = ++idx] = u;
}
}
void pre()
{
lg[1] = 0;
for (int i = 2; i <= idx; ++ i)
lg[i] = lg[i>>1]+1;
for (int i = 1; i <= idx; ++ i)
minn[i][0] = dfn[i];
for (int i = 1; (1<<i) <= idx; ++ i)
for (int j = 1, ub = idx-(1<<i)+1; j <= ub; ++ j){
if (dep[minn[j][i-1]] > dep[minn[j+(1<<(i-1))][i-1]])
minn[j][i] = minn[j+(1<<(i-1))][i-1];
else minn[j][i] = minn[j][i-1];
}
}
int dis(int x, int y)
{
if (fs[x] > fs[y]) swap(x, y);
int len = lg[ed[y]-fs[x]+1];
return dep[x]+dep[y]-(min(dep[minn[fs[x]][len]], dep[minn[ed[y]-(1<<len)+1][len]])<<1);
}
void get_zx(int u, int fa)
{
int son = 0;
sz[u] = 1;
for (register int i = g.h[u]; i; i = g.nxt[i]){
int v = g.v[i];
if (vis[v] || v == fa) continue;
get_zx(v, u);
chkMax(son, sz[v]);
sz[u] += sz[v];
}
chkMax(son, siz-sz[u]);
if (son < mx) mx = son, zx = u;
}
void calc(int u, int fa, int rt)
{
a[rt].add(dpt[u]);
for (register int i = g.h[u]; i; i = g.nxt[i]){
int v = g.v[i];
if (v == fa || vis[v]) continue;
dpt[v] = dpt[u]+1;
calc(v, u, rt);
}
}
int update(int u)
{
if (b[u].sz == 1 && lit[u]){
return b[u].top();
}
else if (b[u].sz >= 2){
int tmp = b[u].top();
b[u].del(tmp);
int ret = b[u].top()+tmp;
b[u].add(tmp);
return ret;
}
return 0;
}
void build(int u)
{
vis[u] = 1;
for (register int i = g.h[u]; i; i = g.nxt[i]){
int v = g.v[i];
if (vis[v]) continue;
siz = sz[v]; mx = inf;
get_zx(v, u);
r[zx] = u;
dpt[v] = 1;
calc(v, u, zx);
if (a[zx].sz) b[u].add(a[zx].top());
build(zx);
}
ans.add(update(u));
}
inline void modify_add(int u, int x)
{
int d = dis(r[u], x);
if (a[u].sz == 0 || a[u].top() <= d){
int tmp = update(r[u]);
if (a[u].sz) b[r[u]].del(a[u].top());
b[r[u]].add(d);
int now = update(r[u]);
ans.del(tmp), ans.add(now);
}
a[u].add(d);
}
inline void modify_del(int u, int x)
{
int d = dis(r[u], x);
if (a[u].top() == d){
int tmp = update(r[u]);
if (a[u].sz) b[r[u]].del(d);
a[u].del(d);
if (a[u].sz) b[r[u]].add(a[u].top());
int now = update(r[u]);
ans.del(tmp), ans.add(now);
}
else a[u].del(d);
}
int main()
{
n = rdi();
g.clear(); litn = n;
for (int i = 1; i <= n; ++ i) lit[i] = 1;
for (int i = 1; i < n; ++i){
int x, y;
x = rdi(); y = rdi();
g.add_undir(x, y);
}
idx = dep[0] = 0; dfs(1, 0); pre();
siz = n; mx = inf;
get_zx(1, 0);
build(zx);
q = rdi();
for (; q--; ){
char opt[1];
scanf("%s", opt);
if (opt[0] == 'G'){
if (litn == 0) puts("-1");
else printf("%d\n", ans.top());
}
else{
int x;
x = rdi();
int tmp = update(x);
lit[x] ^= 1;
if (lit[x]){
for (int i = x; r[i]; i = r[i])
modify_add(i, x);
++ litn;
}
else{
for (int i = x; r[i]; i = r[i])
modify_del(i, x);
-- litn;
}
int now = update(x);
ans.del(tmp), ans.add(now);
}
}
return 0;
}
/*
2
1 2
9
G
C 1
G
C 2
G
C 1
G
C 2
G
*/