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