2019杭电多校 HDU - 6686 Rikka with Travels 树形dp求两部分直径+后缀

题目链接：https://vjudge.net/problem/HDU-6686

题意：找出两个不相交的路径，经过的点数分别为l1,l2，问有多少对(l1, l2)

题解：我们通过枚举每一条边，把树分成两部分，直径分别为 l1、l2，那么也就是说对于（x，y），x在[1, l1]内，y可以取到 l2，x在[1, l2]内，y可以取到 l1，所以我们对应每个位置标记个最大值，最后求个后缀最大即可。那么问题就在于怎么求两部分的直径。

首先明确对于节点u，直径可能经过u，也有可能在子树中。因此:

dp[u][i] 表示节点u到叶节点路径第 i 多的点数

zj[u][i] 表示u的子树中第 i 长的直径

具体实现看一下代码吧

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e5 + 10;
int n;
vector<int> v[N]; 
int dp[N][3], pre[N];
int id[N][3];
int zj[N][2], zjid[N][2];
int maxx[N];
void init() {
	for(int i = 1; i <= n + 1; i++) {
		v[i].clear();
		maxx[i] = 0;
	}
}
void dfs1(int u, int fa) {
	dp[u][0] = dp[u][1] = dp[u][2] = 0;
	id[u][0] = id[u][1] = id[u][2] = 0;
	zj[u][0] = zj[u][1] = 0;
	zjid[u][0] = zjid[u][1] = 0;
	int to;
	int cnt;
	for(int i = 0; i < v[u].size(); i++) {
		to = v[u][i];
		if(to == fa) continue;
		dfs1(to, u);
		if(dp[to][0] >= dp[u][0]) {
			dp[u][2] = dp[u][1];
			id[u][2] = id[u][1];
			
			dp[u][1] = dp[u][0];
			id[u][1] = id[u][0];
			
			dp[u][0] = dp[to][0];
			id[u][0] = to;
		} else if(dp[to][0] >= dp[u][1]) {
			dp[u][2] = dp[u][1];
			id[u][2] = id[u][1];
			
			dp[u][1] = dp[to][0];
			id[u][1] = to;
		} else if(dp[to][0] >= dp[u][2]) {
			dp[u][2] = dp[to][0];
			id[u][2] = to;
		}
		cnt = max(dp[to][0] + dp[to][1] - 1, zj[to][0]);
		if(cnt >= zj[u][0]) {
			
			zj[u][1] = zj[u][0];
			zjid[u][1] = zjid[u][0];
			
			zj[u][0] = cnt;
			zjid[u][0] = to;
		} else if(cnt >= zj[u][1]) {
			zj[u][1] = cnt;
			zjid[u][1] = to;
		}
	}
	dp[u][0]++;
	dp[u][1]++;
	dp[u][2]++;
}
void dfs2(int u, int fa, int premax) { // premax传递的是还没有加u这个子树时的直径 
	int to;
	int l, r;
	for(int i = 0; i < v[u].size(); i++) {
		to = v[u][i];
		if(to == fa) continue;
		r = dp[to][0] + dp[to][1] - 1;
		r = max(r, zj[to][0]);
		if(to == id[u][0]) {
			l = dp[u][1] + max(dp[u][2], pre[u]) - 1;
			pre[to] = max(dp[u][1], pre[u]) + 1;	
		} else if(to == id[u][1]) {
			l = dp[u][0] + max(dp[u][2], pre[u]) - 1;
			pre[to] = max(dp[u][0], pre[u]) + 1;
		} else {
			l = dp[u][0] + max(dp[u][1], pre[u]) - 1;
			pre[to] = max(dp[u][0], pre[u]) + 1;
		}
		
		if(zjid[u][0] == to) {
			l = max(l, zj[u][1]);
		} else {
			l = max(l, zj[u][0]);
		}
		l = max(l, premax);
		maxx[l] = max(maxx[l], r);
		maxx[r] = max(maxx[r], l);
	//	cout << to << " " <<l << " " << r << endl;
		dfs2(to, u, l);
	}
}
int main() {
	int x, y;
	int T;
	ll ans;
	scanf("%d", &T);
	while(T--) {+
		scanf("%d", &n);
		init();
		for(int i = 1; i < n; i++) {
			scanf("%d %d", &x, &y);
			v[x].push_back(y);
			v[y].push_back(x);
		}
		dfs1(1, 0);
		pre[1] = 1;
		dfs2(1, 0, 0);
		ans = 0;
		for(int i = n - 1 ; i >= 1; i--) {
			maxx[i] = max(maxx[i], maxx[i + 1]);
			ans += maxx[i];
	//		cout << i << " " << maxx[i] << endl;
		}
		printf("%lld\n", ans);
	}
	return 0;
} 
/*
5
7
1 2
1 3
2 4
2 5
4 6
5 7
*/