程序设计思维与实践 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

思路

解法

step0 应用链式前向星存储图，edges用来作为边池，u,v,w,nxt分别代表边的两个顶点，w为边的权重，nxt指该点所连边的下一条边。dis1、dis2记录树中各点到直径两端点的距离
step1 将边录入链式前向星中，令dis1[1]=0，从1点开始，对图进行深搜，不断将新到一点的距离记录在dis中（利用到这个新到点的起始点中记录的距离加上改变长度），若比d[maxn]还大，则更新maxn为新到点，遍历结束后，maxn中记录直径的一个端点
step2 从该端点出发重复类似step1的步骤，找到直径的另一个端点，同时将各点到第一个端点的距离记录在dis1中
step3 从第二个端点出发重复类似step1的步骤，将各点到第二个端点的距离记录在dis2中
step4 将各点的dis1、dis2中较大者输出。

错误

1、注意题目给出，输入文件包含多组输入数据！一开始按照一组数据来做，一直出错。
2、改成多组数据输入时，因为若用数组来存储，则对每组数据都需要先将存储数组初始化，这时内存会超出范围。考虑到用数组存储时，该数组为稀疏矩阵，所以可以考虑用链表或链式前向星等存储方式代替。

积累

存储方式

遍历方式

代码

#include<iostream>
#include<queue>
#include<string.h>
using namespace std;
const int maxn=10010;
int maxv=1;
int dis1[maxn],dis2[maxn];
int N; 
struct edge
{
	int u,v,w,nxt;
}edges[2*maxn];
int head[maxn],tot;
void init(int n)
{
	tot=0;
	for(int i=0;i<=n;i++)
		head[i]=-1;
}
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++;
}
void dfs(int n1,int n2,int *d)
{
	for(int i=head[n2];i!=-1;i=edges[i].nxt)
	{
		int v=edges[i].v;
		int w=edges[i].w;
		if(v!=n1)
		{
			d[v]=d[n2]+w;
			if(d[maxv]<d[v]) maxv=v;
			dfs(n2,v,d); 
		}
	}
}
int main()
{
	int c,w;
	while(cin>>N)
	{
	init(N);
	for(int i=2;i<=N;i++)
	{
		cin>>c>>w;
		addEdge(i,c,w);
		addEdge(c,i,w);
	}
	dis1[1]=0;
	dfs(0,1,dis1);
	dis1[maxv]=0;
	dfs(0,maxv,dis1);
	dis2[maxv]=0;
	dfs(0,maxv,dis2);
	for(int i=1;i<=N;i++)
	{
		cout<<max(dis1[i],dis2[i])<<endl;
	}		
	}

	return 0;
 }