bzoj 4871: [Shoi2017]摧毁“树状图” [树形DP]

4871: [Shoi2017]摧毁“树状图”

题意：一颗无向树，选两条边不重复的路径，删去选择的点和路径剩下一些cc，求最多cc数。

---

update 5.1 : 刚刚发现bzoj上这个做法被hack了....以后再想一下别的做法吧

一开始以为这是三合一，写了x=2和x=1. 后来才明白...人家给出的本来就是最优...你自己再求也无所谓

x=0的树形DP没有想出来，感觉很不好处理。

题解是对边进行树形DP

对于有向边(p:(u,v)),(f(p), g(p), d(p))分别表示从v出发走一条，在v子树中走一条经过v，v子树 的最大cc数

注意这个cc数不考虑u所在cc

转移和我一开始想的传统树形DP类似

当时我就卡在了如何更新答案的地方：

1. 两条路径不相交，(d[i]+d[i oplus 1])，或者一个点的两个子树任选+1
2. 两条路径交于一点，从这一点走出三条或四条

(d[i]+d[i oplus 1])这一个转移很巧妙！利用了边是有方向的！

然后我WA了半天结果是被n=1 hack了....

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
#define fir first
#define sec second
const int N = 2e5+5;
inline int read() {
    char c=getchar(); int x=0,f=1;
    while(c<'0' || c>'9') {if(c=='-')f=-1; c=getchar();}
    while(c>='0' && c<='9') {x=x*10+c-'0'; c=getchar();}
    return x*f;
}

int n, u, v;
struct edge{int v, ne;} e[N];
int cnt=1, h[N];
inline void ins(int u, int v) {
	e[++cnt] = (edge){v, h[u]}; h[u] = cnt;
	e[++cnt] = (edge){u, h[v]}; h[v] = cnt;
}

int f[N], g[N], d[N], vis[N], de[N];
struct meow{
	int a, b, c, d;
	meow():a(0), b(0), c(0), d(0){}
};
void dp(int p) {
	if(vis[p]) return; 
	vis[p] = 1;
	int u = e[p].v, child = 0;
	meow t;
	for(int i=h[u];i;i=e[i].ne) if(i != (p^1)) {
		child++;
		dp(i);
		d[p] = max(d[p], d[i]);
		if(f[i] >= t.a) t.b = t.a, t.a = f[i];
		else if(f[i] > t.b) t.b = f[i];
	}
	f[p] = max(t.a, 1) + child - 1;
	g[p] = max(t.a, 1) + max(t.b, 1) + child - 2;
	d[p] = max(d[p], g[p]);
}

void solve() {
	int ans = 0;
	for(int i=2; i<=cnt; i++) dp(i), dp(i^1), ans = max(ans, d[i] + d[i^1]);
	for(int u=1; u<=n; u++) {
		meow t; meow r;
		for(int i=h[u];i;i=e[i].ne) {
			if(f[i] >= t.a) t.d = t.c, t.c = t.b, t.b = t.a, t.a = f[i];
			else {
				if(f[i] >= t.b) t.d = t.c, t.c = t.b, t.b = f[i];
				else {
					if(f[i] >= t.c) t.d = t.c, t.c = f[i];
					else if(f[i] > t.d) t.d = f[i];
				}
			}
			if(d[i] >= r.a) r.b = r.a, r.a = d[i];
			else if(d[i] > r.b) r.b = d[i];
		}
		ans = max(ans, r.a + r.b + 1);
		ans = max(ans, t.a + t.b + t.c + de[u] - 3);
		ans = max(ans, t.a + t.b + t.c + t.d + de[u] - 4);
	}
	printf("%d
", ans);
}

int main() {
	freopen("in", "r", stdin);
	int T=read(), x=read();
	while(T--) {
		n=read(); 
		cnt = 1; for(int i=1; i<=n; i++) h[i] = de[i] = 0;
		if(x==1) read(), read();
		if(x==2) read(), read(), read(), read();
		for(int i=1; i<n; i++) u=read(), v=read(), ins(u, v), de[u]++, de[v]++;
		for(int i=1; i<=cnt; i++) vis[i] = 0, f[i] = g[i] = d[i] = 0;
		if(n == 1) {puts("0"); continue;}
		solve();
	}
	return 0;
}