题意:
求树上每个点能走出的最远距离
思路:
树形DP
第一次dfs求出u点向子树方向走的最长路(dis1)和次长路(dis2)
状态转移 tmp[u] = edge + dis1[v]
第二次dfs求出u点向父亲方向走的最长路
状态转移 tmp[u] = edge + ( u == fa1[f] ) ? dis2[f] : dis1[f]
代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define M 10010
int n,tot;
int head[M],fa1[M],fa2[M];
__int64 dis1[M],dis2[M];
struct edge
{
int v,w,next;
}ed[M*2];
void dfs1(int u,int f)
{
int i,v;
__int64 tmp;
for(i=head[u];~i;i=ed[i].next)
{
v=ed[i].v;
if(v!=f)
{
dfs1(v,u);
tmp=dis1[v]+ed[i].w;
if(tmp>dis2[u])
{
dis2[u]=tmp;
fa2[u]=v;
}
if(dis2[u]>dis1[u])
{
swap(dis2[u],dis1[u]);
swap(fa2[u],fa1[u]);
}
}
}
}
void dfs2(int u,int f)
{
int i,v;
__int64 tmp;
for(i=head[u];~i;i=ed[i].next)
{
v=ed[i].v;
if(v==f)
{
tmp=ed[i].w;
if(fa1[f]==u) tmp+=dis2[f];
else tmp+=dis1[f];
if(tmp>dis2[u])
{
dis2[u]=tmp;
fa2[u]=f;
}
if(dis2[u]>dis1[u])
{
swap(dis2[u],dis1[u]);
swap(fa2[u],fa1[u]);
}
break;
}
}
for(i=head[u];~i;i=ed[i].next)
{
v=ed[i].v;
if(v!=f) dfs2(v,u);
}
}
int main()
{
int i,v,w;
while(~scanf("%d",&n))
{
memset(head,-1,sizeof(head));
tot=0;
memset(dis1,0,sizeof(dis1));
memset(dis2,0,sizeof(dis2));
for(i=2;i<=n;i++)
{
scanf("%d%d",&v,&w);
ed[tot].v=v; ed[tot].w=w; ed[tot].next=head[i]; head[i]=tot++;
ed[tot].v=i; ed[tot].w=w; ed[tot].next=head[v]; head[v]=tot++;
}
dfs1(1,-1);
dfs2(1,-1);
for(i=1;i<=n;i++) printf("%I64d\n",max(dis1[i],dis2[i]));
}
return 0;
}