HDU 2196 Computer（树形DP）

题目链接：点击打开链接

题意：求树上每个点到其他点的最长距离。

思路：典型的树形DP，分析发现，当前节点的最大距离只有两种来源，一种是它到叶子节点的最大距离，一种是经过父亲节点的最大距离，不过，如果父亲节点的最大距离等于父亲节点经过当前节点的最大距离，方案是不可取的，即此处取的是父亲节点到叶子节点不经过当前点的最大距离 + 到当前点的距离，因为要求不经过当前点，所以我们求当前节点到叶子节点的最大距离时还要维护一个次长距离（注意，当前节点到本子树叶子节点的最长路和次长路不能为同一子节点出发的）。

// HDU 2196 Computer.cpp 运行/限制：31ms/1000ms
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
//前向星存树
struct node {
	int to, weight, next;
}edge[20010];
int cnt, head[10005];
int dp[10005][3];
void add(int x, int to, int dis) {
	cnt++;
	edge[cnt].to = to;
	edge[cnt].weight = dis;
	edge[cnt].next = head[x];
	head[x] = cnt;
}
void dfs1(int x, int fa) {//求dp[i][0]和dp[i][1]
	for (int i = head[x]; i; i = edge[i].next) {
		int y = edge[i].to;
		if (y == fa) continue;
		dfs1(y, x);
		if (dp[y][0] + edge[i].weight > dp[x][1]) {
			dp[x][1] = dp[y][0] + edge[i].weight;
			if (dp[x][0] < dp[x][1]) swap(dp[x][0], dp[x][1]);
			//if (dp[y][1] + edge[i].weight > dp[x][1]) {//多加了这段代码，一直wa，最长路和次长路不能为同一子节点出发的，所以不能加
			//	dp[x][1] = dp[y][1] + edge[i].weight;
			//}
		}
	}
}
void dfs2(int x,int fa) {//求dp[i][2]
	for (int i = head[x]; i; i = edge[i].next) {
		int y = edge[i].to;
		if (y == fa) continue;
		if (dp[y][0] + edge[i].weight == dp[x][0]) {
			dp[y][2] = max(dp[x][1], dp[x][2]) + edge[i].weight;
		}
		else {
			dp[y][2] = max(dp[x][0], dp[x][2]) + edge[i].weight;
		}
		dfs2(y, x);
	}
}
int main(){
	int n;
	int x, dis;
	while (scanf("%d", &n) != EOF) {
		cnt = 0;
		memset(head, 0, sizeof(head));
		for (int i = 2; i <= n; i++) {
			scanf("%d%d", &x, &dis);
			add(i, x, dis);
			add(x, i, dis);
		}
		memset(dp, 0, sizeof(dp));
		dfs1(1, 0);
		dfs2(1, 0);
		for (int i = 1; i <= n; i++) {
			printf("%d\n", max(dp[i][0], dp[i][2]));
		}
	}
	return 0;
}