HDU2196 Computer (树形DP入门)

题意：

一棵$n$个点的树，询问每个点距离其他点的最远距离。

分析：

距离这个点的最远距离点无非有两种情况，一种在这个点的子树中，一种是在这个点的父亲节点往上走的点。
设$dp[u][0]$表示$u$节点子树中的最远距离
$dp[u[1]$表示$u$节点子树中的次远距离
$dp[u][2]$表示不在$u$节点子树中的最远距离
$u$是$v$的父节点，则有状态转移方程：
$I.$叶子节点$\to$根节点
$dp[u][1]=dp[u][0]，dp[u][0]=dp[v][0]+w，if(dp[x][0]<dp[v][0]+w)$
$dp[u][1]=max(dp[u][1],dp[v][0]+w)，if(dp[x][0]<=dp[v][0]+w)$
$II.$根节点$\to$叶子节点
$dp[v][2]=max(dp[u][2]+w,dp[u][1]+w)，if(dp[v][0]+w=dp[u][0])$
$ps:$这个方程表示$v$在$u$的最长链上
$dp[v][2]=max(dp[u][2]+w,dp[u][1]+w)，if(dp[v][0]+w≠dp[u][0])$
这个则否。
$I.$中是更新每个节点的子树的最大值和次大值。
$II.$有了子树的最大值和次大值就能完全确定最大值。
得用次大值的原因是：有可能一个节点就在他的父亲节点的最长链上，所以就不能根据父亲节点的最长链来转移而是父亲的次长链。

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> pii;
typedef pair<LL, LL> pll;

const int maxn  = 1e4 + 5;
const int maxm  = 100 + 5;
const int inf   = 0x3f3f3f3f;
const LL  mod   = 1e9 + 7;//19260817
const double pi = acos(-1.0); 

int n, x, cnt, val, head[maxn], dp[maxn][3];

//dp[i][0] i节点子树最大值
//dp[i][1] i节点子树次大值
//dp[i][2] i节点非子树最大值 

struct node{
	int to, val, next;
}edge[maxn << 1];

void addedge(int u, int v, int w){
	edge[++cnt].to = v;
	edge[cnt].val = w;
	edge[cnt].next = head[u];
	head[u] = cnt;
}

void dfs(int x, int f){
	for(int i = head[x]; i; i = edge[i].next){
		int v = edge[i].to, w = edge[i].val;
		if(v == f) continue;
		dfs(v, x);
		if(dp[x][0] < dp[v][0] + w){
			dp[x][1] = dp[x][0];
			dp[x][0] = dp[v][0] + w;
		}else{
			dp[x][1] = max(dp[x][1], dp[v][0] + w);
		}
	}
}

void dfs1(int x, int f){
	for(int i = head[x]; i; i = edge[i].next){
		int v = edge[i].to, w = edge[i].val;
		if(v == f) continue;
		if(dp[v][0] + w == dp[x][0]){
			dp[v][2] = max(dp[x][2] + w, dp[x][1] + w);
		}else{
			dp[v][2] = max(dp[x][2] + w, dp[x][0] + w); 
		}
		dfs1(v, x);
	}
}

int main(){
	while(~scanf("%d", &n)){
		memset(head, 0, sizeof head);
		memset(dp, 0, sizeof dp);
		cnt = 0;
		for(int i = 2; i <= n; i++){
			scanf("%d %d", &x, &val);
			addedge(i, x, val), addedge(x, i, val);
		}
		dfs(1, -1);
		dfs1(1, -1);
		for(int i = 1; i <= n; i++) printf("%d\n", max(dp[i][0], dp[i][2]));
	}
    return 0;
}