[学习笔记] dsu on tree

发布时间 2023-05-25 18:14:01作者: Crazy!!!
  • 适用于动态维护子树信息

  • 流程
    类似树链剖分定义重儿子,轻重链。
    先便历轻边子树,不保存信息(删)。
    最后便历重儿子子树,保存重儿子信息。
    再加入轻儿子子树即可得到子树信息了。
    复杂度:一个点被加/删的次数为轻边数( \(logn\) )。

  • code

点击查看代码
void solve(int u, int fa, bool op) {
	In[u] = ++Time; dfn[Time] = u; 
	for(int i = head[u]; i; i = nxt[i]) {
		int v = to[i]; if(v == fa || v == son[u]) continue;
		solve(v, u, 0);
	}
	if(son[u]) solve(son[u], u, 1);
	Add(a[u]);
	for(int i = head[u]; i; i = nxt[i]) {
		int v = to[i]; if(v == fa || v == son[u]) continue;
		for(int t = In[v]; t <= Out[v]; t++) {
			Add(a[dfn[t]]);
		}
	}
	if(!op) {Del();}
	Out[u] = Time;
}