题目链接:题目传送门
题意:求树上每个点的最远距离是多少
思路:
对一棵树的一个结点求他的最远距离只有两种情况,要么往他的儿子结点一路走到叶子结点,要么向他的父节点往走,然后在往下一直走到叶节点,(但需要注意的时,一但你往下朝子节点走了,就不能再往父节点走了)。
对于第一种情况,第一个搜索,直接回溯搜索。(从树的下部到上部得出结果)
而对于第二种情况,需要用到第一种情况的数据,所以需要进行玩第一个搜索,才可以进行第二割搜索。第二种情况主要分为两种,要么就是往父节点之后继续再往上,要么到了父节点之后开始一路往下走到叶结点。(从树的上部到下部依次得到结果)
#include<cstdio>
#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
#include<vector>
using namespace std;
#define maxn 10000+10
int n;
int f[maxn];
int g[maxn],p[maxn];
//f[u]=max(f[v]+w(u,v))
//g[u]=w(u,p[u])+max(g[p[u]],f[v]+w(p[u],v)),v是u的兄弟
struct node
{
int v,w;
node(int a,int b)
{
v=a;
w=b;
}
};
vector<node>e[maxn*2];
int dfs1(int u,int pre)
{
int len=e[u].size();
for(int i=0;i<len;i++)
{
int v=e[u][i].v;
if(v!=pre)
{
f[u]=max(f[u],dfs1(v,p[v]=u)+e[u][i].w);
}
}
return f[u];
}
void dfs2(int u,int pre)
{
int t=0;
g[u]=g[pre];
int len=e[pre].size();
for(int i=0;i<len;i++)
{
int v=e[pre][i].v;
if(v==p[pre])continue;
if(v==u)t=e[pre][i].w;
else g[u]=max(g[u],f[v]+e[pre][i].w);
}
g[u]+=t;
len=e[u].size();
for(int i=0;i<len;i++)
{
int v=e[u][i].v;
if(v!=pre)
dfs2(v,u);
}
}
int main()
{
while(~scanf("%d",&n))
{
for(int i=1;i<=n;i++)e[i].clear();
f[1]=g[1]=p[1]=0;
for(int u=2;u<=n;u++)
{
int w,v;
scanf("%d%d",&v,&w);
e[u].push_back(node(v,w));
e[v].push_back(node(u,w));
f[u]=g[u]=p[u]=0;
}
dfs1(1,0);
dfs2(1,0);
for(int i=1;i<=n;i++)
{
printf("%d\n",max(f[i],g[i]));
}
}
return 0;
}