题意:
实验室里原先有一台电脑(编号为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
思路:
分析:
要确定最长距离,肯定是到两“头”距离才可能最长,如何确定树的两头呢?转化为求树的直径,确定树的两‘头’
具体做法:
假设树的最长路的两个叶子节点为v1,v2,从任意一点u出发走到的最 远的点一定是(v1,v2)中的一点,然后再从v1或者v2出发走到的最 远点一定是v2或者v1。由此经过两次遍历就能找到树的两个叶子。
再分别从v1,v2点遍历到要求的点,取max即得到最长路径。
注意点:
这里我是用bfs进行遍历,用vis数组标记每个点是否到达过,由于多次调用bfs,每次使用后都要及时将vis数组置0,记录路径长度的数组len也同理,需及时更新
总结:
1.先是将问题进行转化,按照求树的直径的方法求出树的两片叶子
2.bfs,dfs遍历的熟练掌握
3.细节上注意,数组置0
代码:
#include<cstdio>
#include<vector>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
#define maxn 10001
struct edge{
int from,to,weight;
edge(int a,int b,int c){from=a;to=b;weight=c;}
};
vector<edge> G[maxn];
bool vis[maxn];
int len1[maxn],len2[maxn],fa[maxn]; //记录路径长度,前面的节点数
queue<int> q;
int N,x,y,v,v1,v2;
void bfs1(int u){ //确定树直径两端点v1,v2
vis[u]=1;
q.push(u);
int f=0;
while(!q.empty()){
int w=q.front();
q.pop(); //删除有标记的点
for(int i=0,n=G[w].size();i<n;i++){
if(vis[G[w][i].to]==0){
fa[G[w][i].to]=fa[w]+G[w][i].weight;
q.push(G[w][i].to);
vis[G[w][i].to]=1;
if(fa[G[w][i].to]>f){f=fa[G[w][i].to];v=G[w][i].to;}
}
}
}
memset(fa,0,sizeof(fa));memset(vis,0,sizeof(vis));
}
void f_bfs2(int u){ //确定路径长度
vis[u]=1;
q.push(u);
while(!q.empty()){
int w=q.front();
q.pop(); //删除有标记的点 从v1开始遍历
for(int i=0,n=G[w].size();i<n;i++){
if(vis[G[w][i].to]==0){
len1[G[w][i].to]=len1[w]+G[w][i].weight;
q.push(G[w][i].to);
vis[G[w][i].to]=1;
}
}
}
memset(vis,0,sizeof(vis));
}
void b_bfs2(int u){ //确定路径长度 从v2开始遍历
vis[u]=1;
q.push(u);
while(!q.empty()){
int w=q.front();
q.pop(); //删除有标记的点
for(int i=0,n=G[w].size();i<n;i++){
if(vis[G[w][i].to]==0){
len2[G[w][i].to]=len2[w]+G[w][i].weight;
q.push(G[w][i].to);
vis[G[w][i].to]=1;
}
}
}
memset(vis,0,sizeof(vis));
}
int main(){
while(scanf("%d",&N)!=EOF){
for(int i=2;i<=N;i++){
scanf("%d%d",&x,&y);
G[i].push_back({i,x,y});
G[x].push_back({x,i,y});
}
bfs1(1); v1=v;
bfs1(v1); v2=v;
f_bfs2(v1); b_bfs2(v2);
for(int i=1;i<=N;i++){
if(len1[i]>len2[i]) printf("%d\n",len1[i]);
else printf("%d\n",len2[i]);
}
memset(len1,0,sizeof(len1));memset(len2,0,sizeof(len2));
for(int i=1;i<=N;i++)
G[i].clear(); //清空
}
return 0;
}