Week6 A - 氪金带东

Week6 A - 氪金带东

实验室里原先有一台电脑(编号为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找到最远点 t
从最远点开始遍历 将每个点离 t 的距离记录进 v1 找出另一个最远点 t
从 t 开始遍历 将每个点离 t 的距离记录进 v2
v1 v2 分别记录每个点到两端的距离
Code

#include<iostream>
#include<vector>
#include<string.h>
#include<algorithm>
using namespace std;
int head[100010];
int vis[100010];
int v1[100010];
int v2[100010];

struct edge{
	int a1;
	int a2;
	int nu;
}edge[100010];
int n=1,t;
void add(int a,int b,int nu){
	edge[n].a1=head[a];
	edge[n].a2=b;
	edge[n].nu=nu;
	head[a]=n++;
}
void dfs(int u,int wa,int *f){
	f[u]=wa;
		if(f[t]<f[u]){
			t=u;
		}
	for(int i=head[u];i!=0;i=edge[i].a1){
		if(!vis[edge[i].a2]){wa+=edge[i].nu;//cout<<u<<' '<<edge[i].a2<<endl;
		
			vis[edge[i].a2]=true;
			dfs(edge[i].a2,wa,f);
			wa-=edge[i].nu;
		}
	}
}
int main(){
    int a=0;
    int u,v,w;
    int nn;
	while(scanf("%d",&a)!=EOF){
	memset(head,0,sizeof(head));
	for(int i=2;i<=a;i++){
		cin>>v>>w;//cout<<i<<'-'<<v<<endl;
		add(i,v,w);
		add(v,i,w);
	}
	memset(vis,0,sizeof(vis)); 
	v1[t]=0;
	vis[1]=true;
	dfs(1,0,v1);
	memset(vis,0,sizeof(vis)); 
	v1[t]=0;
	vis[t]=true;
	dfs(t,0,v1);
	memset(vis,0,sizeof(vis)); 
	v2[t]=0;
	vis[t]=true;
	dfs(t,0,v2);
	
	for(int i=1;i<=a;i++){
		cout<<max(v1[i],v2[i])<<endl;
	}	
	}
	
}