程序设计思维与实践 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找到直径的一个端点，然后从这个端点出发找到另外一个直径端点，并且同时更新各点到此端点的距离。然后从另外一个端点再进行一次dfs，更新各点到端点的距离，取各点到两个端点的最大值。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>

using namespace std;

//每一个点的最大距离一定是到达直径上的某一个端点 

struct node{

long long to;

	long long w;

};

vector<node> p[10001];

long long dis[10001];

long long max_len,s;

//len是当前点距离起始点的距离 

void dfs(long long u,long long fa,long long len){

	//如果max_len<len说明还能延伸

	//直到找到不能延伸的点

	//该点就是树的直径的端点之一 

	if(max_len<=len){

		s=u;//最终记录下一个直径端点 

		max_len=len;

	}

	long long Len=p[u].size();

	for(long long i=0;i<Len;i++){

		long long v=p[u][i].to;

		if(v==fa)continue;

		dfs(v,u,len+p[u][i].w);

		//更新v点到起始点的最长距离 

		dis[v]=max(dis[v],len+p[u][i].w);

	}

}

 

int main(){

	long long n;

	while(cin>>n){

		for(long long i=0;i<=n;i++){

			p[i].clear();

		}

		for(long long i=2;i<=n;i++){

			long long y,w;

			cin>>y>>w;

			node t;

			t.to=y;

			t.w=w;

			p[i].push_back(t);

			t.to=i;

			p[y].push_back(t);

		}

		memset(dis,0,sizeof(dis));

		max_len=0;

		s=0;

		dfs(1,-1,0);
		dfs(s,-1,0);

		dfs(s,-1,0);

		for(long long i=1;i<=n;i++){

			cout<<dis[i]<<endl;
		}
	}
	return 0;
}