[ZJOI2007]Hide 捉迷藏(动态点分治/点分树)

动态点分治模板题。

题意

一棵树,每个点可以是黑色或者白色。

有一些询问,一种是修改一个点的颜色,另一种查询整棵树中距离最远的两个黑点的距离。

思路

还有一种做法是括号序列,详见某岛?,这里不赘述了,因为并不会并不是我写这一题的重点。

所以考虑建出点分树。然后在每个点维护两个堆:

  1. u作为重心的管辖区域内,所有黑点到u在点分树上的父亲的距离。
  2. 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
*/
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值