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

问题分析

图的存储——链式前向星

使用数组来模拟链表
使用一个结构体数组Edge[M] 记录每条边的信息以及next来指向下一 条边的（下标）， 使用一个数组Head[N] 表示以每个顶点为头的第一条边在结构体数组 中的下标，Head初始化为-1
插入一条边的时候，插入位置为链表的头部：Edge数组的next指向 Head[u]然后Head[u]更新为新加入边的下标

树的直径

树的直径被定义为树中任意两点之间距离的最大值，一定是某两个叶子节点间的距离。
如何求树的直径
从树中任选一个点开始遍历这棵树，找到一个距离这个点最远的叶子， 然后再从这个叶子开始遍历，找到离这个叶子最远的另一个叶子，他俩之间的距离就是树的直径。
需要做两次bfs或dfs。

解题思路

这道题要找某个节点x所能到达的最长路径，因为最长路径的端点一定是直径的一个端点。即若直径为（v1, v2），则节点x所能到达的最长路径为（x, v1）或 (x, v2) 。
因此此题可以通过第一次从节点1遍历找到树的直径的一个端点v1，然后从v1进行第二次遍历找到端点v2，同时更新树中任意节点到v1的距离dis1。之后从v2进行第三次遍历，更新树中任意节点到v2的距离dis2。
dis1[i]与dis2[i]中的最大值即为节点i所能到达的最长路径。
注意
题目中说的是会输入多组数据，while(scanf("%d",&n) != EOF)
记录直径端点的变量mx，初始值要为1 (因为代码中是从节点1开始遍历的）。

代码实现

#include<stdio.h>
#define N 10050
#include<algorithm>
using namespace std;

struct edge
{
	int u,v,w;
	int net;
};

int n=0;
int index = 0;
int head[N];
edge e[50000];
int re[N];
int *len1, *len2;
int mx;

void init()
{
	for(int i = 0; i <=n; i++)
	    re[i] = 0;
}


void add(int u, int v, int w)
{
	e[index].u = u;
	e[index].v = v;
	e[index].w = w;
	e[index].net = head[u];
	head[u] = index;
	index++;
}

void dfs(int start, int d[])
{
	for(int i=head[start]; i!=-1; i=e[i].net)
	{
		edge x = e[i];
		if(re[x.v] == 0)
		{//未曾访问过 
			re[x.v] = 1;//标记到达 
			d[x.v] = d[start] + x.w;//更新距离 
			if(d[mx] < d[x.v]) mx = x.v;//最远点 
			dfs(x.v, d);
		}
	}
}

int main()
{
	while(scanf("%d",&n) != EOF)
{
	for(int i=0; i<=n; i++) head[i] = -1;
	index = 0;
	for(int i=2; i<=n; i++)
	{
		int v,w;
		scanf("%d%d",&v,&w);
		add(i, v, w);
		add(v, i, w);
	}
	
	len1 = new int[n+1];
	len2 = new int[n+1];
	//求直径的一个端点
	len1[1]=0;
	init();
	re[1] = 1;
	mx=1;
	dfs(1,len1);
	
	//求直径的另一个端点,计算从一个端点到各个点的距离 
	len1[mx] = 0;
	init(); 
	re[mx] = 1;
	dfs(mx,len1);
	
	//计算从另一个端点到各个点的距离
	len2[mx] = 0;
	init();
	re[mx] = 1;
	dfs(mx, len2); 
	
	for(int i=1; i<=n; i++) printf("%d\n",max(len1[i], len2[i])) ;
}
return 0;
 }