程序设计思维与实践 Week6 作业A氪金带东树的直径的应用

题意：

依次输入图中的点以及边权等信息，最后输出每个点在图中所能到达的最远的路线的长度。例如所给的样例：

input:

输入文件包含多组测试数据。对于每组测试数据，第一行一个整数N (N<=10000)，接下来有N-1行，每一行两个数，对于第i行的两个数，它们表示与i号电脑连接的电脑编号以及它们之间网线的长度。网线的总长度不会超过10^9，每个数之间用一个空格隔开。

output:

对于每组数据输出n行，分别表示N个点所能到达的最远的地方。

思路：

对于每个点能到达的最远的地方，必然是图中最长的也就是直径上的两端点上的任一点。因此变成了每个点到达直径两端点的最远长度。如果从每个点开始跑一遍dfs，复杂度过高。可以反过来想，即从两端点开始跑，跑到任意点的时候记录距离，然后取这两个距离中较长的那个即为最远的距离。其中找直径可以任选一个点，跑一遍dfs，记录下能到达的最远的那个点，一定是直径两端点其中的一个，再从这个端点跑一遍dfs，记录下能到达的最远的点便是两端点中的另外一个，再跑第二遍的时候可以记录下到达每个点的距离。然后再从另一个端点跑一遍dfs，记录下跑到每个点的距离，然后在第二遍的基础上更新所能到达的点的更大的距离。最后输出更新后的距离即可得到答案。其中在记录图的信息的时候可以选择链式前向星，即用一个结构体数组分别记录下边的信息，包括出点，入点，边权，下一个边，每次增加一个边后，可以用一个变量tot记录下当前边的数量。一个head数组记录每个点的后面所链接的邻接点的信息，实现链表的功能。其中双向边需要同时加入两个边。

代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <vector>
using namespace std;

struct edge{
	int from,to,value,next;
};

int N;  //电脑个数
int head[10010];   //点的邻接 
edge P[20010];  //记录边的信息 

bool vis[10010];  //标记 
int length[10010]; //表示距离  
int tot=0;  //记录边数 
int l1,l2;  //最远的两个点
int maxlength=0;
int length2[10010];


void addedge(int f, int t, int v){
	P[tot].from=f;
	P[tot].to=t;
	P[tot].value=v;
	
	P[tot].next=head[f];
	head[f]=tot;
	tot++;
}


//找最远的点 
void dfs1(int x){
	vis[x]=1;
	for(int i=head[x];i!=-1 ;i=P[i].next) {
		int m=P[i].to;  //所到达的点 
		if(!vis[m]){
			vis[m]=1;
			length[m]=length[x]+P[i].value;
			if(length[m]>maxlength){
				maxlength=length[m];
				l1=m;
			}
			dfs1(m);	
		}
	}
}

void dfs2(int x){
	vis[x]=1;
	for(int i=head[x];i!=-1 ;i=P[i].next) {
		int m=P[i].to;  //所到达的点 
		if(!vis[m]){
			vis[m]=1;
			length[m]=length[x]+P[i].value;
			if(length[m]>maxlength){
				maxlength=length[m];
				l2=m;
			}
			dfs2(m);	
		}
	}
}

void dfs3(int x){
	vis[x]=1;
	for(int i=head[x];i!=-1 ;i=P[i].next) {
		int m=P[i].to;  //所到达的点 
		if(!vis[m]){
			vis[m]=1;
			length2[m]=length2[x]+P[i].value;
			if(length2[m]>length[m]){
				length[m]=length2[m];
			}
			dfs3(m);	
		}
	}
}

int main(){
	
	while(cin>>N){
			//初始化标记数组 
		for(int i=1;i<=N;i++){
			vis[i]=0;
			length[i]=0;
			length2[i]=0;
			head[i]=-1;
		}
		for(int i=0; i<N;i++){
			P[i].next=-1;
		}
		
		 
		int a,b; 
		tot=0;
		for(int i=2;i<=N;i++){
			scanf("%d%d",&a,&b);  //邻接点及边权 
			addedge(i,a,b); 
		    addedge(a,i,b);			
		}
		
		maxlength=0;
		dfs1(1);
		
		maxlength=0;
		for(int i=1; i<=N;i++){
			length[i]=0;
			vis[i]=0;
		}
	    dfs2(l1);
	    
		for(int i=1; i<=N;i++){
			vis[i]=0;
		}
		dfs3(l2);
		
		for(int i=1; i<=N;i++){
			cout<<length[i]<<endl;
		}
	}

	return 0;
}