题意:
给出一棵树,以及每个节点之间的距离,求每个节点能到达的最长距离,这里的树是广义上的树。
题解:
设置这样的状态dp[i][3]为什么,因为我们需要对树的儿子部分求最长路和父亲部分求最长路,那么0表示儿子的最长路,1表示儿子的次长路,2表示父亲的最长路。那么就可以两次搜索dp。
代码附上注释了。
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>
#include<math.h>
using namespace std;
#define oo 0x3f3f3f3f
#define maxn 10005
int dp[maxn][3];//0表示儿子最长路,1儿子表示次长路,2表示父亲最长路
struct EDGE
{
int v,d,next;
}E[maxn<<1];
int head[maxn],tol;
void add_edge(int u,int v,int d)
{
E[tol].v=v;
E[tol].d=d;
E[tol].next=head[u];
head[u]=tol++;
}
void inst()
{
memset(head,-1,sizeof head);
tol=0;
}
void Down(int u,int fa)
{
for(int i=head[u];i!=-1;i=E[i].next)
{
int v=E[i].v;
if(v==fa) continue;
Down(v,u);
if(dp[u][0]<E[i].d+dp[v][0])//跟最长比大的话替换
{
dp[u][1]=dp[u][0];
dp[u][0]=E[i].d+dp[v][0];
}
else if(dp[u][1]<E[i].d+dp[v][0])//比最长路短,跟次长路比
dp[u][1]=E[i].d+dp[v][0];
}
}
void Up(int u,int fa)
{
int len;
for(int i=head[u];i!=-1;i=E[i].next)
if(E[i].v==fa)
{
len=E[i].d;
break;
}
if(fa!=0)//根节点要特判,因为根节点不用计算
{
if(dp[u][0]+len==dp[fa][0])//如果这个点在父亲儿子部分的最长路上
//那么对应的最长路肯定在父亲儿子部分的次长路和父亲部分的最长路上
{
if(dp[fa][1]>dp[fa][2])
dp[u][2]=dp[fa][1]+len;
else
dp[u][2]=dp[fa][2]+len;
}
else//如果这个点不在父亲儿子部分的最长路上
//那么对应的最长路肯定在父亲儿子部分的最长路和父亲部分的最长路上
{
if(dp[fa][0]>dp[fa][2])
dp[u][2]=dp[fa][0]+len;
else
dp[u][2]=dp[fa][2]+len;
}
}
for(int i=head[u];i!=-1;i=E[i].next)
if(E[i].v!=fa)
Up(E[i].v,u);
}
int main()
{
int n,u,v,d;
while(scanf("%d",&n)!=EOF)
{
inst();
memset(dp,0,sizeof dp);
for(int i=2;i<=n;i++)
{
scanf("%d %d",&v,&d);
add_edge(i,v,d);
add_edge(v,i,d);
}
Down(1,0);
Up(1,0);
for(int i=1;i<=n;i++)
printf("%d\n",max(dp[i][0],dp[i][2]));
}
return 0;
}