dsu on tree (树上启发式合并) 详解

一直都没出过算法详解，昨天心血来潮想写一篇，因而 dsu on tree 它来了

一、前置技能

1.链式前向星（vector 建图）

2.dfs 建树

3.剖分轻重链，轻重儿子

重儿子	一个结点的全部儿子中拥有最多子树的儿子
轻儿子	一个结点的全部儿子中不是重儿子的儿子
重边	父亲与重儿子的连边
轻边	父亲与轻儿子的连边
重链	一堆重边链接而成的链
轻链	一堆轻边链接而成的链

二、什么是 dsu on tree（树上启发式合并） ？

dsu on tree 其实就是个优雅的暴力算法，和它一块儿共被称为优雅暴力的算法还有莫队
所谓优雅的暴力大概是指：“优雅思想，暴力的操做”
例如莫队咱们知道它是将整个区间分块，再将询问的区间排序，最后暴力的维护全部询问的区间
其中 "整个区间分块，询问的区间排序" 为优雅的思想，而 "暴力的维护全部询问的区间" 为暴力的操做
由于须要将询问的区间排序，咱们就须要先将询问的区间保存下来，也就是要离线
dsu on tree 和莫队相似，也须要离线（它们同属于静态算法）

dsu on tree 优雅的思想：

对于以 u 为根的子树

①. 先统计它轻子树(轻儿子为根的子树)的答案，统计完后删除信息

②. 再统计它重子树(重儿子为根的子树)的答案 ，统计完后保留信息

③. 而后再将重子树的信息合并到 u上

④. 再去遍历 u 的轻子树，而后把轻子树的信息合并到 u 上

⑤. 判断 u 的信息是否须要传递给它的父节点（u 是不是它父节点的重儿子）

dsu on tree 暴力的操做

dsu on tree 暴力的操做体现于统计答案上（不一样的题目统计方式不同）

三、dsu on tree 的过程演示及代码

1.图示

• 1 的重儿子为 2，轻儿子为 3

• 2 的重儿子为 4，轻儿子为 5

• 3 没有重儿子，没有轻儿子

• 4 的重儿子为 6，没有轻儿子

• 5 的重儿子为 7，没有轻儿子

• 6 没有重儿子，没有轻儿子

• 7 没有重儿子，没有轻儿子

为了更好观看，咱们将节点与其重儿子的连线描红

咱们从根节点1进入，先找1的轻儿子，发现3，进入3

3没有别的儿子能够进入了，因而统计3的信息

统计完后即将返回父节点 1

由于1-3的边没有被描红边、3不是1的重儿子（不传递3的信息），因此删除3的信息再返回 1

发现1没有别的轻儿子了，就找重儿子，发现2，进入2

进入2后，再找2的轻儿子，发现5，进入5

发现5没有轻儿子了，就找重儿子，发现7，进入 7

7 没有别的儿子能够进入了，因而统计 7 的信息

统计完后即将返回父节点 5

由于边5-7 有被描红边、7是5的重儿子，因此保留7的信息直接返回 5（传递7的信息的给5）

5 全部儿子都进入过了，因而统计 5 的信息

统计完后即将范围父节点 2

由于边2-5 没有被描红边、5不是2的重儿子，因此删除5的信息再返回 2

发现2没有其它轻儿子了，就找重儿子，发现4，进入4

发现4没有其它轻儿子了，就找重儿子，发现6，进入6

6 没有别的儿子能够进入了，因而统计 6 的信息

统计完后即将返回父节点 4

由于边4-6 有被描红边，6是4的重儿子，因此保留6的信息直接返回 4（传递6的信息的给4）

4 全部儿子都进入过了，因而统计 4 的信息

统计完后即将返回父节点 2

由于边2-4 有被描红边，4是2的重儿子，因此保留4的信息直接返回2（传递4的信息的给2）

2 全部儿子都进入过了，因而统计 2 的信息

2 接受了4传递的信息，可是并无接受5传递给它的信息（被删除了）

因而 2 再进入5（轻儿子），统计一遍以 5 为根的子树的信息，再将该信息合并到 2上

统计完后 2 后即将返回父节点 1

由于边1-2 有被描红边，2是1的重儿子，因此保留2的信息直接返回1（传递2的信息的给1）

1 全部儿子都进入过了，因而统计 1 的信息

1 接受了2传递的信息，可是并无接受3传递给它的信息（被删除了）

因而 1 再进入3（轻儿子），统计一遍以 3 为根的子树的信息，再将该信息合并到 1 上

至此，整个 dsu on tree 的过程结束

2.代码

struct Edge{
	int nex , to;
}edge[N << 1];
int head[N] , TOT;
void add_edge(int u , int v) // 链式前向星建图
{
	edge[++ TOT].nex = head[u] ;
	edge[TOT].to = v;
	head[u] = TOT;
}
int sz[N];   // sz[u] 表示以 u 为根的子树大小 
int hson[N]; // hson[u] 表示 u 的重儿子 
int HH;      // HH 表示当前根节点的重儿子 
void dfs(int u , int far)
{
	sz[u] = 1;
	for(int i = head[u] ; i ; i = edge[i].nex) // 链式前向星 
	{
		int v = edge[i].to;
		if(v == far) continue ;
		dfs(v , u); 
		sz[u] += sz[v];  
		if(sz[v] > sz[hson[u]]) hson[u] = v; // 选择 u 的重儿子 
	}
}
void calc(int u , int far , int val) // 统计答案 
{
	if(val == 1) ...; // val = 1，则添加信息 
	else ...;         // val = -1，则删除信息 
	......	
    for(int i = head[u] ; i ; i = edge[i].nex)
    {
        int v = edge[i].to;
        if(v == far || v == HH) continue ; // 若是 v 是当前根节点的重儿子，则跳过
        calc(v , u , val);
    }
} 
void dsu(int u , int far , int op)  // op 等于0表示不保留信息，等于1表示保留信息 
{
	for(int i = head[u] ; i ; i = edge[i].nex)
	{
		int v = edge[i].to;
		if(v == far || v == hson[u]) continue ; // 若是 v 是重儿子或者父亲节点就跳过 
		dsu(v , u , 0);     // 先遍历轻儿子 ，op = 0 :轻儿子的答案不作保留 
	}
	if(hson[u]) dsu(hson[u] , u , 1) , HH = hson[u];
	// 轻儿子都遍历完了，若是存在重儿子，遍历重儿子（事实上除了叶子节点每一个点都必然有重儿子）
	// op = 1 ， 保留重儿子的信息 
	// 当前是以 u 为根节点的子树，因此根节点的重儿子 HH = hson[u]
	calc(u , far , 1); // 再次遍历轻儿子统计答案
        HH = 0;			   // 遍历结束 ，即将返回父节点，因此取消标记 HH 
	if(!op) calc(u , far , -1); // 若是 op = -1，则 u 对于它的父亲来讲是轻儿子，不须要将信息传递给它的父亲 
}

4.经典例题讲解

题目编号	题目连接	题解连接
CF600E	https://codeforces.com/problemset/problem/600/E	http://www.javashuo.com/article/p-zfgilixw-nx.html
CF570D	https://codeforces.com/problemset/problem/570/D	http://www.javashuo.com/article/p-amgiyhws-nx.html
CF208E	https://codeforces.com/problemset/problem/208/E	http://www.javashuo.com/article/p-eyebfsoc-nx.html
CF246E	https://codeforces.com/problemset/problem/246/E	http://www.javashuo.com/article/p-wgeyxxxw-nx.html
CF1009F	https://codeforces.com/problemset/problem/1009/F	http://www.javashuo.com/article/p-zcmargri-nx.html
CF375D	https://codeforces.com/problemset/problem/375/D	http://www.javashuo.com/article/p-dphgfsxw-nx.html
wannafly Day2 E	https://ac.nowcoder.com/acm/contest/4010/E?&headNav=acm	http://www.javashuo.com/article/p-ukyhjnjg-nx.html
ccpc2020长春站F题	https://codeforces.com/gym/102832/problem/F	http://www.javashuo.com/article/p-xsfsjfat-nx.html
洛谷P4149	https://www.luogu.com.cn/problem/P4149	http://www.javashuo.com/article/p-rsramweu-nx.html

5.难题进阶

这是道较难的题，据说这也是 dsu on tree 的发明人专门为这个算法出的题

题目编号	题目连接	题解连接
CF741D	https://codeforces.com/contest/741/problem/D	http://www.javashuo.com/article/p-pthqnupk-nx.html

┏┛ ┻━━━━━┛ ┻┓  
    ┃　　　　　　┃  
    ┃　　　━　　　      ┃  
    ┃　┳┛　  ┗┳　    ┃  
    ┃　　　　　　       ┃  
    ┃　　　┻　　　    ┃  
    ┃　　　　　　      ┃  
    ┗━┓　　　┏━━━┛  
      ┃　　　┃   神兽保佑  
      ┃　　　┃   代码无BUG！  
      ┃　　　┗━━━━━━━━━┓  
      ┃　　　　　　　    ┣┓  
      ┃　　　　         ┏┛  
      ┗━┓ ┓ ┏━━━┳ ┓ ┏━┛  
          ┃ ┫ ┫   ┃ ┫ ┫  
          ┗━┻━┛   ┗━┻━┛