题目:
实验室里原先有一台电脑(编号为1),最近氪金带师咕咕东又为实验室购置了N-1台电脑,编号为2到N。每台电脑都用网线连接到一台先前安装的电脑上。但是咕咕东担心网速太慢,他希望知道第i台电脑到其他电脑的最大网线长度,但是可怜的咕咕东在不久前刚刚遭受了宇宙射线的降智打击,请你帮帮他。
提示: 样例输入对应这个图,从这个图中你可以看出,距离1号电脑最远的电脑是4号电脑,他们之间的距离是3。 4号电脑与5号电脑都是距离2号电脑最远的点,故其答案是2。5号电脑距离3号电脑最远,故对于3号电脑来说它的答案是3。同样的我们可以计算出4号电脑和5号电脑的答案是4.
输入:
输入文件包含多组测试数据。对于每组测试数据,第一行一个整数N (N<=10000),接下来有N-1行,每一行两个数,对于第i行的两个数,它们表示与i号电脑连接的电脑编号以及它们之间网线的长度。网线的总长度不会超过10^9,每个数之间用一个空格隔开。
输出:
对于每组测试数据输出N行,第i行表示i号电脑的答案 (1<=i<=N).
样例输入:
5
1 1
2 1
3 1
1 1
样例输出:
3
2
3
4
4
在解决这道题前,要先了解如何求树的直径。求树的直径需要用两次bfs/dfs,第一次从任一点出发,找到离该点最远的点,然后再从找到的最远点开始,进行一次bfs/dfs,找到离该点最远的点,这两个点就是树的直径的两个端点。
但是本题不是找直径,是找某个节点x所能到达的最长路径,那么最长路径的第二个节点就是直径的两个端点之一,因为不能确定是到哪个端点,所以还需要一次bfs/dfs。
本题的数据结构用邻接链表。
以下是完整代码:
#include<iostream>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;
struct edge{
int v,w;
edge(int vx,int wx)
{
v=vx;
w=wx;
}
edge()
{
}
};
int reach[11000];//bfs过程中用来标记是否已经访问过
long long ans1[11000],ans2[11000],d[11000],ans[11000];//前三个用于bfs过程中存储距离,ans用来存储最后答案
int N;
vector<edge> a[11000];
void bfs(int x)
{
for(int i=0;i<=N;i++)
{
reach[i]=0;
d[i]=0;
}
queue<int> q;
q.push(x);
int temp;
reach[x]=1;
while(!q.empty())
{
temp=q.front();
q.pop();
for(int i=0;i<a[temp].size();i++)
{
edge next=a[temp][i];
if(reach[next.v]==0)
{
d[next.v]=d[temp]+next.w;
q.push(next.v);
reach[next.v]=1;
}
}
}
}
void bfs1(int x)//求出第二个端点,同时求出所有点到第一个端点的距离
{
for(int i=0;i<=N;i++)
{
reach[i]=0;
ans1[i]=0;
}
queue<int> q;
q.push(x);
int temp;
reach[x]=1;
while(!q.empty())
{
temp=q.front();
q.pop();
for(int i=0;i<a[temp].size();i++)
{
edge next=a[temp][i];
if(reach[next.v]==0)
{
ans1[next.v]=ans1[temp]+next.w;
q.push(next.v);
reach[next.v]=1;
}
}
}
}
void bfs2(int x)//同时求出所有点到第二个端点的距离
{
for(int i=0;i<=N;i++)
{
reach[i]=0;
ans2[i]=0;
}
queue<int> q;
q.push(x);
int temp;
reach[x]=1;
while(!q.empty())
{
temp=q.front();
q.pop();
for(int i=0;i<a[temp].size();i++)
{
edge next=a[temp][i];
if(reach[next.v]==0)
{
ans2[next.v]=ans2[temp]+next.w;
q.push(next.v);
reach[next.v]=1;
}
}
}
}
void solve()
{
bfs(1);
int maxv=0;
int k=0;
for(int i=1;i<=N;i++)//求出直径的一个端点
{
if(maxv<d[i])
{
maxv=d[i];
k=i;
}
}
int max1=k;
bfs1(max1);
maxv=0;
k=0;
for(int i=1;i<=N;i++)//求出直径的第二个端点
{
if(maxv<ans1[i])
{
maxv=ans1[i];
k=i;
}
}
int max2=k;
bfs2(max2);
for(int i=1;i<=N;i++)//把ans1和ans2比较,取较大值放入ans数组
{
if(ans1[i]<ans2[i])
{
ans[i]=ans2[i];
}
else
{
ans[i]=ans1[i];
}
}
for(int i=1;i<=N;i++)
{
cout<<ans[i]<<'\n';
}
}
int main()
{
while(cin>>N)
{
for(int i=2;i<=N;i++)
{
int v,w;
cin>>v>>w;
a[i].push_back(edge(v,w));
a[v].push_back(edge(i,w));
}
solve();
for(int i=1;i<=N;i++)//把之前的数据清空
{
a[i].clear();
}
}
}