树的直径解题

问题描述

实验室里原先有一台电脑(编号为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

关于链式前向星

• 使用数组来模拟链表
• 相比较邻接链表而言，边在链式前向星中存储的相对顺序与输入顺序
相反，其他差异不大。
• 使用一个结构体数组 Edge[M] 记录每条边的信息以及 next 来指向下一
条边的（下标）
• 使用一个数组 Head[N] 表示以每个顶点为头的第一条边在结构体数组
中的下标，Head 初始化为 -1
• 插入一条边的时候，插入位置为链表的头部：Edge 数组的 next 指向
Head[u] 然后 Head[u] 更新为新加入边的下标
存储方式：使用一个结构体数组记录图中的所有边，结构体里有三个成员：u，v，w
分别表示这个边关联的两个顶点和边权

树的直径

树的直径被定义为树中任意两点之间距离的最大值
做法：
树的直径一定是某两个叶子之间的距离
• 从树中任选一个点开始遍历这棵树，找到一个距离这个点最远的叶子，然后再从这个叶子开始遍历，找到离这个叶子最远的另一个叶子，他俩之间的距离就是树的直径。
• 两次遍历即可求的树的直径。
• 遍历可以用DFS也可以用BFS，找到距离起点最远的叶子结点就好

问题分析

假设树的最长路的两个叶子节点为v1, v2，从任意一点 u 出发走到的最远的点一定是（v1,v2）中的一点，然后再从 v1 或者 v2 出发走到的最远点一定是 v2 或者 v1。由此经过两次遍历就能找到最长路径。
本题要解决找某个节点 x 所能到达的最长路径问题，则这个节点的最长路径为到v1的路径或者到v2的路径，从v1，v2出发进行遍历所有点求距离，加上原先确定直径则总共是三次dfs遍历
此处在dfs遍历时加for循环确保遍历每个点，并不断更新最长路径，与之前数据结构求固定两点路径不同，固定两点路径确定否则成环，在遍历到目的点时即停止遍历，此处dfs遍历了所有点

代码段出错点

初始化使存储边信息的数组下标为0，同时使head数组的初始指向也为零，并作为跳出循环的判断，这样edges【0】的边以及和它相关边的信息就无法被利用。

#include<iostream>
#include<cstdio> 
using namespace std;

struct edge{
	int u,v,w,nxt;
}edges[20020];

int head[10010],tot;
void addEdge(int u,int v,int w)
{
	edges[tot].u=u;
	edges[tot].v=v;
	edges[tot].w=w;
	edges[tot].nxt=head[u];
	head[u]=tot;
	tot++;
}

bool vis1[10010],vis2[10010],vis3[10010];
int sum1,sum2,sum3; 
int dis1[10010],dis2[10010],dis3[10010];
int v1,v2; 
void dfs1(int u)
{
	vis1[u]=true;
	for(int i=head[u];i!=0;i=edges[i].nxt)
	{
		if(!vis1[edges[i].v])//检查临接点 
		{
			vis1[edges[i].v]=true;
			dis1[edges[i].v]=dis1[u]+edges[i].w;
			if(dis1[edges[i].v]>sum1)
			{
				sum1=dis1[edges[i].v];
				v1=edges[i].v;
			}
			dfs1(edges[i].v);
		}
	}
}

void dfs2(int u)
{
	vis2[u]=true;
	for(int i=head[u];i!=0;i=edges[i].nxt)
	{
		if(!vis2[edges[i].v])//检查临接点 
		{
			vis2[edges[i].v]=true;
			dis2[edges[i].v]=dis2[u]+edges[i].w;
			if(dis2[edges[i].v]>sum2)
			{
				sum2=dis2[edges[i].v];
				v2=edges[i].v;
			}
			dfs2(edges[i].v);
		}
	}
}

void dfs3(int u)
{
	vis3[u]=true;
	for(int i=head[u];i!=0;i=edges[i].nxt)
	{
		if(!vis3[edges[i].v])//检查临接点 
		{
			vis3[edges[i].v]=true;
			dis3[edges[i].v]=dis3[u]+edges[i].w;
			dfs3(edges[i].v);
		}
	}
}

int main()
{
	int v,w,n;
	while(scanf("%d",&n)!=EOF)
	{
		for(int i=0;i<10010;i++)
		{
			head[i]=dis1[i]=dis2[i]=dis3[i]=0;
			vis1[i]=vis2[i]=vis3[i]=false;
		} 
		sum1=sum2=sum3=0;
		tot=1;//报错点 
		for(int i=2;i<=n;i++)
		{
			cin>>v>>w;
			addEdge(i,v,w);
			addEdge(v,i,w);
		}
		dfs1(1);
		dfs2(v1);
		dfs3(v2);
		for(int i=1;i<=n;i++)
			cout<<max(dis2[i],dis3[i])<<endl;
	}
	return 0;
}