题目
实验室里原先有一台电脑(编号为1),最近氪金带师咕咕东又为实验室购置了N-1台电脑,编号为2到N。每台电脑都用网线连接到一台先前安装的电脑上。但是咕咕东担心网速太慢,他希望知道第i台电脑到其他电脑的最大网线长度,但是可怜的咕咕东在不久前刚刚遭受了宇宙射线的降智打击,请你帮帮他。
提示: 样例输入对应这个图,从这个图中你可以看出,距离1号电脑最远的电脑是4号电脑,他们之间的距离是3。 4号电脑与5号电脑都是距离2号电脑最远的点,故其答案是2。5号电脑距离3号电脑最远,故对于3号电脑来说它的答案是3。同样的我们可以计算出4号电脑和5号电脑的答案是4.
Input
输入文件包含多组测试数据。对于每组测试数据,第一行一个整数N (N<=10000),接下来有N-1行,每一行两个数,对于第i行的两个数,它们表示与i号电脑连接的电脑编号以及它们之间网线的长度。网线的总长度不会超过10^9,每个数之间用一个空格隔开。
Output
对于每组测试数据输出N行,第i行表示i号电脑的答案 (1<=i<=N).
Sample Input
5
1 1
2 1
3 1
1 1
Sample Output
3
2
3
4
4
思路:
这是一个求树的最大直径的问题。如果按每个点进行一次dfs或者bfs的话,复杂度为O(n^2),复杂度太高。我们注意到,每个点到其他点的最大距离就是它到树的最大直径的两个端点的距离最大值,所以我们先求树的最大直径,并从每个点到最大直径dfs求距离即可。如何求树的最大直径呢,先从任意一个点开始,dfs求离它最远距离的那个叶子,再dfs从这个叶子找到离它最远的叶子,这两个叶子之间就是树的最大直径。
代码:
#include<iostream>
#include<cstring>
#include<cstdio>
#define MAXN 10005
#define MAXM 20010
using namespace std;
int head[MAXN],tot;
int vis[MAXN];
int len1[MAXN];//记录每个点到目标点的距离
int len2[MAXN];
int len3[MAXN];
int x;
int maxn=0;//不要把这个写在dfs函数里面
struct Edge
{
int u;
int v;
int w;
int nxt;
}edge[MAXM];
void init(int n)
{
tot=0;
memset(head,-1,sizeof(head));
}
void addEdge(int u,int v,int w)
{
edge[tot].u=u;
edge[tot].v=v;
edge[tot].w=w;
edge[tot].nxt=head[u];
head[u]=tot;
tot++;
}
void dfs(int u,int *s)
{
vis[u]=1;
for(int i=head[u];i!=-1;i=edge[i].nxt)
{
if(vis[edge[i].v]==0)
{
s[edge[i].v]=s[u]+edge[i].w;
if(s[edge[i].v]>maxn)
{
x=edge[i].v;
maxn=s[edge[i].v];
}
dfs(edge[i].v,s);
}
}
}
int main()
{
int n;
while(~scanf("%d",&n))
{
memset(vis,0,sizeof(vis));
memset(len1,0,sizeof(len1));
memset(len2,0,sizeof(len2));
memset(len3,0,sizeof(len3));
init(n);
for(int i=2;i<=n;i++)
{
int v,w;
scanf("%d %d",&v,&w);
addEdge(i,v,w);
addEdge(v,i,w);
}
dfs(1,len1);
maxn=0;
int v1=x;
memset(vis,0,sizeof(vis));
dfs(v1,len2);
maxn=0;
int v2=x;
memset(vis,0,sizeof(vis));
dfs(v2,len3);
maxn=0;
for(int i=1;i<=n;i++)
{
printf("%d\n",max(len2[i],len3[i]));
}
}
return 0;
}