毁灭计划
题目链接: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;
}