【ybt高效进阶5-4-3】最长距离（树形DP）

最长距离

题目链接：ybt高效进阶5-4-3

题目大意

给你一棵树，边有距离，对于每个点，要你求跟它距离最远的点到它的距离。

思路

看到这种题，因为是最大值所以换根 DP 不太好搞，所以我们就考虑直接树形 DP。

那这样的话距离要么就是直接向下，要么就是向上再向下。
直接向下很好搞，接着问题是第二个。
那你考虑可能是父亲已经搞好，然后要搞儿子，那可能是从父亲的那条路再下一个位置向上走，也可能这里就是向上到向下转折的地方。
那转折的话就要再找一条路径，不难想到一开始跑直接向下的时候顺便维护一下第二大的向下路径，然后就可以转移了。

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#define ll long long

using namespace std;

struct node {
	int x, to, nxt;
}e[10001];
int n, fa[10001], y, z;
int le[10001], KK, pl[100001];
ll f[10001][3];

void csh() {
	KK = 0;
	memset(f, 0, sizeof(f));
	memset(le, 0, sizeof(le));
}

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

void dfs1(int now, int father) {
	ll maxn = 0, maxx = 0;
	for (int i = le[now]; i; i = e[i].nxt)
		if (e[i].to != father) {
			dfs1(e[i].to, now);
			if (maxn <= f[e[i].to][0] + e[i].x) {//跑出最大次大（向下）
				maxx = maxn;
				maxn = f[e[i].to][0] + e[i].x;
				pl[now] = e[i].to;
			}
			else if (maxx < f[e[i].to][0] + e[i].x) {
				maxx = f[e[i].to][0] + e[i].x;
			}
			else if (maxx < f[e[i].to][1] + e[i].x) {
				maxx = f[e[i].to][1] + e[i].x;
			}
		}
	f[now][0] = maxn;
	f[now][1] = maxx;
}

void dfs2(int now, int father) {
	for (int i = le[now]; i; i = e[i].nxt)//跑向上的
		if (e[i].to != father) {//要么是原本至今好的在多下一格要么就是从这里开始下
			if (e[i].to == pl[now]) f[e[i].to][2] = max(f[now][2] + e[i].x, f[now][1] + e[i].x);
				else f[e[i].to][2] = max(f[now][2] + e[i].x, f[now][0] + e[i].x);
			dfs2(e[i].to, now);
		}
}

int main() {
	while (scanf("%d", &n) != EOF) {
		for (int i = 2; i <= n; i++) {
			scanf("%d %d", &y, &z);
			add(y, i, z);
		}
		
		dfs1(1, 0);
		dfs2(1, 0);
		
		for (int i = 1; i <= n; i++)
			printf("%lld\n", max(f[i][0], f[i][2]));
		
		csh();
	}
	
	return 0;
}