HDU2196树形dp

1.前言

今天考试考了一道树形dp的题目，我发现我对于树形dp是个白痴，于是我挑了一道网上都说是水题的题目做，然后愉快地调了一个下午。我果然是个蒟蒻。

2.题目思路

f[x]表示x在以它为根的子树中可以到达的最远距离

se[x]表示x在以它为根的子树中可以到达的次远距离

bj[x]表示x在以它为根的子树中到达最远距离走的儿子节点

g[x]表示x通过它的父节点可以到达的最远距离

然后弄两个dfs搞一搞。

dp的方程看代码好了。

3.代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<climits>
#include<iomanip>
#include<vector>
#include<cmath>
#include<algorithm>
using namespace std;
int n,m;
int f[10005],se[10005],g[10005],bj[10005];
vector<int>lu[10005],lon[10005];
void dfs(int x,int las){//处理f,se和bj
	int i,sz=lu[x].size();
	for(i=0;i<sz;i++)
		if(lu[x][i]!=las){
			dfs(lu[x][i],x);
			if(f[lu[x][i]]+lon[x][i]>=f[x]){
				se[x]=f[x];f[x]=f[lu[x][i]]+lon[x][i];
				bj[x]=lu[x][i];
			}
			else if(f[lu[x][i]]+lon[x][i]>se[x])se[x]=f[lu[x][i]]+lon[x][i];
		}
}
void ds(int x,int las){//处理g，要么通过父节点的父节点到更远的远方，要么通过兄弟节点
	int i,sz=lu[x].size();
	for(i=0;i<sz;i++)
		if(lu[x][i]!=las){
			if(lu[x][i]==bj[x])	g[lu[x][i]]=max(se[x]+lon[x][i],g[x]+lon[x][i]);
			else g[lu[x][i]]=max(f[x]+lon[x][i],g[x]+lon[x][i]);
			ds(lu[x][i],x);
		}
}
int main()
{
   	int i,j,x,y;
   	while(scanf("%d",&n)!=EOF){
	   	for(i=1;i<=n;i++){lu[i].clear();lon[i].clear();g[i]=0;f[i]=0;bj[i]=0;se[i]=0;}
	   	for(i=1;i<=n-1;i++){
	   		scanf("%d%d",&x,&y);
	   		lu[x].push_back(i+1);lon[x].push_back(y);
	   		lu[i+1].push_back(x);lon[i+1].push_back(y);
	   	}
	   	dfs(1,0);ds(1,0);
	   	for(i=1;i<=n;i++){printf("%d\n",max(g[i],f[i]));}
   	}
   	return 0;
}