链式前向星、树的直径应用(week6作业)

A - 氪金带东

1.题目

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

5
1 1
2 1
3 1
1 1

3
2
3
4
4

2.解题思路

存储图可以使用邻接矩阵邻接表等等。这里使用链式前向星存储。他的思想是用数组模拟链表。

定义边的结构体，其中包含边的终点v、边的权重w、同一个起点的下一条边的编号next。并且用head数组存储每一个点延伸出去的第一条边的编号。当在目前的图中加入一条边时，除了设置该边的终点权重，还要将该边的起点的head值设为本条边的编号，也就是这里默认当插入一条边时，就让该边作为head的第一条边。

又本题不是找图的最长路径，而是找每个节点所能到达的最长路径。这题可以借助树的直径。树的直径的求法（摘自教学ppt）：

找出树的直径节点v1 v2后，不需要遍历每一个节点并循环该节点到v1、v2的值找到最大值输出，这样复杂度太高。反向思考，我们可以求出v1、v2到每个节点的长度。注意当我们求树的直径的时候，第二遍遍历是遍历v1到每一个节点，这时候就可以将v1到每个节点的距离存储下来。此时只要从v2在遍历到每个节点，存储每个节点到v2的长度。最后判断两个长度输出较长的那个。

遍历可以dfs也可以bfs，这里采用dfs。

还要注意：
因为是无向图，所以边的数组长度需要开组合数*2，插入边也要插两次；
每次遍历完图一遍都要充值vis数组及有关参数。

3.c++代码

#include<stdio.h>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
using namespace std;

//边的编号从0开始，head的编号从1开始 
const int maxn=1e4+5;
int len1[maxn]={0};
int maxV=0,maxL=0;
bool vis[maxn]={false};
int head[maxn];
int tot=0;//head[u]表示以u为头的第一条边在结构体数组edges中的下表 

struct edge
{
	int v;//终节点 
	int w;//权重
	int next;// 同一起点的下一条边 
}edges[100000000];//记录所有边的信息
//无向图 

void init(int n)
{
	tot=0;//tot是edges的下标 
	for(int i=0;i<maxn;i++)
	{
		head[i]=-1; 
		
	 } 
 } 
 //tot是目前该条边的编号 
void addEdge(int _u,int _v,int _w)
{
	//添加一条边后要更新edges数组和head数组
	//这里每次插入一条边都让这条边为以_u为顶点的所有边中的第一条边
	edges[tot].v=_v;
	edges[tot].w=_w;
	edges[tot].next=head[_u];//
	head[_u]=tot;//tot为目前边的下标 
	tot++; 
}

	
void initVis();
void find(int u);
int main()
{
	int n;
	while(scanf("%d",&n)!=EOF)
	{
		init(n);//记得写！！ 
		int v,w;
		for(int i=0;i<n-1;i++)
		{
			cin>>v>>w;
			addEdge(i+2,v,w);
			addEdge(v,i+2,w);
		}		
		//找到直径的两端点
		int l=0,r=0;
		initVis();
		find(1);l=maxV;//最左端点 
	
		initVis();
		int *tmp=new int[n+1];
		find(l);r=maxV;for(int i=1;i<=n;i++)tmp[i]=len1[i];//最右 

		initVis();
		find(r);
		for(int i=1;i<=n;i++)
		{
			if(tmp[i]>len1[i])cout<<tmp[i]<<endl;
			else cout<<len1[i]<<endl;
		}
	}
	return 0;
}

//重置各项数据 
void initVis()
{
	maxV=0;maxL=0;
	for(int i=0;i<maxn;i++)
	{
		vis[i]=false;
		len1[i]=0;
	}
} 

void find(int u)//找到距离u最远的点 
{
	vis[u]=true;
	for(int i=head[u];i!=-1;i=edges[i].next)
	{
		//maxV记录当前最远的点，maxL为最远长度 
		if(!vis[edges[i].v])
		{
			vis[edges[i].v]=true;
			len1[edges[i].v]=edges[i].w+len1[u];
			if(len1[edges[i].v]>maxL)
			{//如果该点才是应当到达的点 
				maxL=len1[edges[i].v];
				maxV=edges[i].v;
			}
			find(edges[i].v);
		}
	}
}