hdu 2196 Computer 经典的树形DP 两次DFS遍历
题意很简单,就是给你一棵树,每条边都有一定的权值,然后让你找到每个点所能走到的最远距离
那么我们可以这样高效的来处理
先以 1 作为根节点进行一次 dfs 遍历,遍历的时候把以 第 i 为根节点往子树方向可以走到的最远距离和次远距离给求出来,且这两个距离是不在同一个分支中的
然后我们进行第二次的从根节点开始dfs遍历,这时候我们要判断的是对于一个节点 i ,除了子树方向上可能有最远距离外,我通过父节点方向是否可以有更加远的距离
,而对于这个操作,我们只要访问父节点所能到达的最远距离,然后接上一段就行了,但是这里会产生一个问题——就是父节点的最远距离可能是会经过当前这个节点
的,因此我们要进行判断当前节点是否在父节点的最远距离的分支上
如果在的话,那么我们要换一个分支,且要是最远的,这个其实就是第一次dfs中求出的与最远路径不在同一分支上的次远路径,我们接上这段进行转移
如果不在的话,那么我们直接接上父节点的最远路径进行转移就行了
代码如下:
#include<iostream>
#include<vector>
using namespace std;
#define N 10001
int n,longest[N],f[N],g[N],dp[N];
//longest[]用来记录当前节点的最远路径所对应的分支编号(因为是邻接表存储的,对于当前节点的判重只要记录下分支编号就行了)
//f[]是当前节点往子树方向所能走过的最远距离,g[]是往子树方向次远距离,dp[]是通过父节点能走过的最远距离
vector<int> son[N],w[N];
int dfs(int root)
{ if(f[root]) return f[root];
int rootsize=son[root].size();
if(!rootsize) return 0; //叶子节点
int est=-1,esti;
int er=-1,eri=-1;
for(int i=0;i<rootsize;i++)
{ if(dfs(son[root][i])+w[root][i]>est)
{ est=f[son[root][i]]+w[root][i];
esti=i;
}
}
longest[root]=esti;
f[root]=est;
for(int i=0;i<rootsize;i++)
{ if(f[son[root][i]]+w[root][i]>er&&i!=esti)
{ er=f[son[root][i]]+w[root][i];
eri=i;
}
}
if(eri!=-1) g[root]=er;
return f[root];
}
void dfs1(int root)
{ int rootsize=son[root].size();
for(int i=0;i<rootsize;i++)
{ if(i==longest[root])
dp[son[root][i]]=max(dp[root],g[root])+w[root][i]; //这里的转移是关键,需要YY一下
else
dp[son[root][i]]=max(dp[root],f[root])+w[root][i];
dfs1(son[root][i]);
}
}
void input()
{ while(scanf("%d",&n)!=EOF)
{ memset(f,0,sizeof(f));
memset(dp,0,sizeof(dp));
memset(g,0,sizeof(g));
memset(longest,-1,sizeof(longest));
for(int i=1;i<=n;i++) son[i].clear(),w[i].clear();
for(int i=2;i<=n;i++)
{ int x1,x2;
scanf("%d%d",&x1,&x2);
son[x1].push_back(i);
w[x1].push_back(x2);
}
f[1]=dfs(1);
dp[1]=0;
dfs1(1);
for(int i=1;i<=n;i++)
printf("%d\n",max(f[i],max(g[i],dp[i])));
}
}
int main()
{ input();
return 0;
}