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;
}