【ybtoj高效进阶 21169】毁灭计划（分类讨论）（树形DP）

毁灭计划

题目链接：ybtoj高效进阶 21169

题目大意

给你一棵树，然后你要找两条无相交边的路径，使得删去这两个路径的点和它相邻的边之后，这个树被分成尽可能多的连通块。

思路

这题就是阴间的分类讨论 DP。
（题解十几页是真的**）

---

首先是设 (f_{i,j}) 是 (i) 的子树内，当前形态是 (j) 的最大连通块数。

然后介绍形态：
(0)：一个端点在当前根 (i)，另一个在子树内。

(1)：不经过 (i)，在 (i) 子树内一条链。

(2)：在 (i) 子树内一条链，经过 (i) 点。

(3)：这个是选两条链，一条是类型 (0) 的一条是类型 (1) 的。

---

然后接着我们考虑两个问题：
如何统计答案？如何转移 (f) 数组。

继续分类讨论，首先考虑解决统计答案。

统计答案一共六种情况。

（后面的 (-[u==1]) 是因为如果不是最根上面就有东西就有一个连通块，如果是跟上面的就没有连通块所以少了一个）

第一种：(f_{u,0}+f_{v,3}-[u==1])

第二种：(f_{u,0}+f_{v,3}-[u==1])

第三种：(f_{u,1}+f_{v,2})

第四种：(f_{u,1}+f_{v,1}-1)（这个减一是上面连一起了少了一个）

第五种：(f_{u,2}+f_{v,1}-[u==1])

第六种：(f_{u,2}+f_{v,2}-[u==1])

---

接着考虑转移。（这里设 (deg_i) 为 (i) 的儿子数）

(0)：
只有一种方法，在儿子的路径上延伸多 (i) 这个儿子。
(f_{v,0}+deg_p-1)

(1)：
方法一：(f_{v,1})

方法二：(f_{v,2}+1)

(2)：
一种方法，(f_{u,0}+f_{v,0}-1)

(3)：
方法一：(f_{u,0}+f_{v,2}-1)

方法二：(f_{u,0}+f_{v,1}-1)

方法三：(f_{u,2}+f_{v,0}-1)

方法四：(f_{v,3}+deg_u-1)

方法五：(f_{v,0}+deg_u+maxn-2)（(maxn) 是之前遍历过的 (v) 中最大的 (f_{v,1},f_{v,2})）

然后注意的是枚举的顺序，统计答案放在最前面，然后 (3) 的要在 (0,2) 之前，(2) 要在 (0) 之间。
（因为用到了，不能用修改之后的，要用修改之前的）

代码

#include<cstdio>
#include<iostream>

using namespace std;

struct node {
	int to, nxt;
}e[1000005];
int T, X, n, x, y;
int le[500005], KK;
int deg[500005], ans;
int f[500005][5];

void csh() {
	KK = 0; ans = 0;
	for (int i = 1; i <= n; i++) le[i] = 0, deg[i] = 0;
}

void add(int x, int y) {
	e[++KK] = (node){y, le[x]}; le[x] = KK;
	e[++KK] = (node){x, le[y]}; le[y] = KK;
}

void DP(int now, int father) {
	f[now][0] = f[now][2] = f[now][3] = deg[now];
	f[now][1] = 1;
	int maxn = 0;
	for (int i = le[now]; i; i = e[i].nxt)
		if (e[i].to != father) {
			DP(e[i].to, now);
			
			ans = max(ans, f[now][3] + f[e[i].to][0] - (now == 1));
			ans = max(ans, f[now][0] + f[e[i].to][3] - (now == 1));
			ans = max(ans, f[now][1] + f[e[i].to][2]);
			ans = max(ans, f[now][1] + f[e[i].to][1] - 1);
			ans = max(ans, f[now][2] + f[e[i].to][1] - (now == 1));
			ans = max(ans, f[now][2] + f[e[i].to][2] - (now == 1));
			
			f[now][1] = max(f[now][1], f[e[i].to][1]);
			f[now][1] = max(f[now][1], f[e[i].to][2] + 1);
			
			f[now][3] = max(f[now][3], f[now][0] + f[e[i].to][2] - 1);
			f[now][3] = max(f[now][3], f[now][0] + f[e[i].to][1] - 1);
			f[now][3] = max(f[now][3], f[now][2] + f[e[i].to][0] - 1);
			f[now][3] = max(f[now][3], f[e[i].to][3] + deg[now] - 1);
			f[now][3] = max(f[now][3], f[e[i].to][0] + deg[now] + maxn - 2);
			
			f[now][2] = max(f[now][2], f[now][0] + f[e[i].to][0] - 1); 
			
			f[now][0] = max(f[now][0], f[e[i].to][0] + deg[now] - 1);
			
			maxn = max(maxn, max(f[e[i].to][1], f[e[i].to][2]));
		}
}

int main() {
//	freopen("destroy.in", "r", stdin);
//	freopen("destroy.out", "w",stdout);
	
	scanf("%d %d", &T, &X);
	while (T--) {
		scanf("%d", &n);
		if (X >= 1) scanf("%d %d", &x, &y);
		if (X >= 2) scanf("%d %d", &x, &y);
		for (int i = 1; i < n; i++) {
			scanf("%d %d", &x, &y);
			add(x, y);
			deg[x]++; deg[y]++;
		}
		for (int i = 2; i <= n; i++) deg[i]--;
		DP(1, 0);
		printf("%d
", ans);
		csh();
	}
	
	return 0;
}