Computer（树的直径）

Computer

一所学校不久前买了第一台电脑（所以这台电脑的ID是1）。近年来，学校购买了N-1新电脑。每台新电脑都连接到一台先前安装的电脑上。学校的管理人员担心网络运行缓慢，希望知道第i台计算机需要发送信号的最大距离si（即到最远计算机的电缆长度）。您需要提供此信息。

提示：示例输入与此图对应。从图中，你可以看到计算机4离1最远，所以s1=3。计算机4和5是距离2最远的，所以s2=2。计算机5是离3最远的，所以s3=3。我们也得到了s4=4，s5=4。
输入
输入文件包含多组测试样例。在每组样例中，第一行中都有自然数n（n<=10000），然后是（n-1）行，其中包含对计算机的描述。第i行包含两个自然数-第i计算机所连接的计算机和用于连接的电缆长度。电缆总长度不超过10^9。输入行中的数字用空格分隔。
输出
对于每组样例，输出n行。第i行第i台计算机的到其他计算机的最大长度Si（1<=i<=n）。
样例输入
5
1 1
2 1
3 1
1 1
样例输出
3
2
3
4
4
提示
示例输入与此图对应。从图中，你可以看到计算机4离1最远，所以s1=3。计算机4和5是距离2最远的，所以s2=2。计算机5是离3最远的，所以s3=3。我们也得到了s4=4，s5=4。

从第一台到第n台分别bfs会超时；用三次bfs可以解决这个问题：1先以任意一台电脑求出距离最远的两台电脑，即树的直径，2然后从树的直径的一端bfs，求出每台电脑到直径端的距离，并记录；3最后从另一端的直径求出每台电脑到这一端的距离，存入；
每台电脑取最大值输出

#include<iostream>
#include<algorithm>
#include<string.h>
#include<queue>
using namespace std;

int n;//m是computer的个数
int a,b,cost;//两个computer的位置以及距离（存的时候两个方向都要存）
char d;
int dis[50005];//端点到每个computer的距离 
int vis[50005]; 
int ed,ans,num;
int head[50005];
struct cow{
	int from;//computer的序号1
	int to;//computer的序号2 
	int next;
	int llng;//距离 
	//friend bool operator<(cow a,cow b){
	//	return a.llng <b.llng ;
	//} 
};
cow e1[100005];
void add(int a,int b,int c){
	e1[num].from =a;
	e1[num].to =b;
	e1[num].llng =c;
	e1[num].next =head[a];
	head[a]=num++;
}
void bfs(int x){
	memset(vis,0,sizeof(vis));
	memset(dis,0,sizeof(dis));
	queue<int> que;
	que.push(x);
	vis[x]=1;
	ans=0;
	while(!que.empty()){
		int u,v,i;
		u=que.front();
		//cout<<"u:"<<u<<endl;
		que.pop();
		for(i=head[u];i!=-1;i=e1[i].next ){
		v=e1[i].to ;
		//cout<<"v:"<<v<<endl; 
		if(!vis[v]){
			
			if(dis[v]<dis[u]+e1[i].llng )
				dis[v]=dis[u]+e1[i].llng;
				vis[v]=1;
				que.push(v);
		//		cout<<"v:::::"<<v<<endl;
			}
		}
	} 
	for(int i=1;i<=n;i++){
	if(dis[i] >ans){
			ans=dis[i];
			ed=i;
		//	cout<<"ans:"<<ans<<" "<<i<<endl;
			}
		} 
	} 

int main(){
	ios::sync_with_stdio(0);
	cin.tie(),cout.tie();
	while(cin>>n){
	int num=1;
	memset(head,-1,sizeof(head));
	for(int i=2;i<=n;i++){
		cin>>b>>cost;
		
		add(i,b,cost);
		add(b,i,cost);
		
	}
	bfs(1);
	bfs(ed);
	int di1[100005];
	int di2[100005];
	for(int i=1;i<=n;i++){
		//cout<<"di1:"<<i<<" "<<dis[i]<<endl;
		di1[i]=dis[i];
		//cout<<ans<<endl;
		//cout<<"%%%"<<endl;
	}
	bfs(ed);
	
	for(int i=1;i<=n;i++){
		//cout<<"di2:"<<i<<" "<<dis[i]<<endl;
		di2[i]=dis[i];
	}
	for(int i=1;i<=n;i++){
	ans=max(di1[i],di2[i]);
	cout<<ans<<endl;
	}
	}
	return 0;
}