题意
实验室里原先有一台电脑(编号为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
思路
本题的思路比较简单,仅需要三次bfs搜索即可。
首先,随便找到一台电脑进行bfs搜索,跟这台电脑距离最远的那台电脑便是这些电脑组成的树的一个叶子
然后对这个树叶进行bfs搜索,距离它最远的电脑也是该树的一个叶子,两者之间的距离便是树的直径,树中所有的节点到这两台电脑的距离的最大者就是该节点到所有电脑的网线的最大长度。
所以需要对另一个叶子电脑进行一次bfs搜索
最后比较每个电脑到这两个电脑的最大者,输出即可(每次进行bfs搜索要将距离存在数组中)
代码
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<queue>
using namespace std;
struct graph
{
long long v;
long long len;
graph* next;
};
queue<long long> q;
void bfs(long long,long long,long long*,graph*,int*);
int main()
{
long long n;
while(~scanf("%lld",&n))
{
graph *a=new graph[n+3];
for(long long i=1;i<=n;i++)
a[i].next=NULL;
int* vis=new int[n+3];
for(long long i=0;i<=n+1;i++)
vis[i]=-1;
for(long long i=2;i<=n;i++)
{
long long b;
long long len;
scanf("%lld%lld",&b,&len);
graph* temp=&a[b],*tmp=new graph;
while(temp->next!=NULL)
temp=temp->next;
tmp->len=len;
tmp->v=i;
tmp->next=NULL;
temp->next=tmp;
graph* temp1=&a[i],*tmp1=new graph;
while(temp1->next!=NULL)
temp1=temp1->next;
tmp1->len=len;
tmp1->v=b;
tmp1->next=NULL;
temp1->next=tmp1;
}
long long *b=new long long[n+3],*c=new long long[n+3],*d=new long long[n+3];
for(long long i=0;i<n+3;i++)
{
b[i]=0;
c[i]=0;
d[i]=0;
}
bfs(1,0,b,a,vis);
long long v1=1,v2=1;
for(long long i=1;i<=n;i++)
if(b[v1]<b[i])
v1=i;
bfs(v1,1,c,a,vis);
for(long long i=1;i<=n;i++)
if(c[v2]<c[i])
v2=i;
bfs(v2,3,d,a,vis);
for(long long i=1;i<=n;i++)
{
if(c[i]>d[i])
cout<<c[i]<<endl;
else
cout<<d[i]<<endl;
}
}
system("pause");
return 0;
}
void bfs(long long u,long long mark,long long b[],graph a[],int vis[])
{
q.push(u);
vis[u]=mark;
while(!q.empty())
{
long long temp=q.front();
q.pop();
graph *t=a[temp].next;
while(t!=NULL)
{
if(vis[t->v]!=mark)
{
vis[t->v]=mark;
q.push(t->v);
b[t->v]=b[temp]+t->len;
}
t=t->next;
}
}
}