训练3.14（J:hdu 2196 Computer）

训练3.14

hdu 2196 Computer
题目：http://acm.hdu.edu.cn/showproblem.php?pid=2196
题意：有n台计算机连成一片，形成一棵无根树。求每台计算机与其他计算机的最远距离。

题解：令U[i],D1[i],D2[i]分别表示结点i向上的最远路径距离，向下的最远路径距离以及向下的次远路径距离。F1[i],F2[i]分别储存结点i向下的最远路径经过的孩子以及向下次远路径经过的孩子。
最远距离将出现在U[i]和D1[i]中。

首先明确：D1，D2可由dfs求出，并存好相应F1，F2。
接着就是U——向上的最远路径了。

假设当前结点为x,设u为x向下最远路径孩子，v为次远路径孩子，那么u和v的向上最远距离可借助x来计算得出。图中加粗部分是u和v的向上最远距离的两种可能情况：
我们现在是在求U[u]或者U[v],再次明确一下。

如图所示（图就凑合着看先）：
左上由于我们事先dfs了所以可以直接算出他的路径长度为val+D2[x];
右上我们可以看见他是经过f,由于整个过程是由上向下进行的（见代码），所以此时的U[f]其实是已经处理好了的，所以路径长度就是U[x]+val。
左上和右上取最大值就是U[u]了。
（val为结点u与结点x的距离）

左下和右下分析与上面类似：左下路径长度为D1[x]+val,右下路径长度为U[x]+val。
左下与右下取最大值就是U[v]。
（val为结点v与结点x的距离）

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
const int maxn=1e+5+5;
struct node{
	int x,val;
	node(int x,int val){
		this->x=x;
		this->val=val;
	}
};
vector<node>G[maxn];
int U[maxn],D1[maxn],D2[maxn];
int F1[maxn],F2[maxn];
int dfs(int x,int f){//返回值为点x向下的最大距离 
	for(int i=0;i<G[x].size();i++){
		int u=G[x][i].x;
		int val=G[x][i].val;	
		if(u==f)continue;
		int tmp=dfs(u,x);
		//更新当前由点x向下的最大距离和次大距离 
		if(tmp+val>D1[x]){ 
			D2[x]=D1[x];D1[x]=tmp+val;
			F2[x]=F1[x];F1[x]=u;
		}
		else if(tmp+val>D2[x]){
			D2[x]=tmp+val;
			F2[x]=u;
		}
	}
	return D1[x];
}
void init(int n){
	for(int x=0;x<=n;x++){
		G[x].clear();
		D1[x]=D2[x]=F1[x]=F2[x]=U[x]=0;	
	}
}
void work(int x,int f){
	int v1,v2;
	v1=v2=0;
	//找到x与最远路径上的孩子的距离v1以及与次远路径上的孩子的距离v2 （向下的最远与次远） 
	for(int i=0;i<G[x].size();i++){
		int u=G[x][i].x;
		if(u==f)continue;
		int val=G[x][i].val;
		if(F1[x]==u)v1=val;
		else if(F2[x]==u)v2=val;
	}
	for(int i=0;i<G[x].size();i++){
		int u=G[x][i].x;
		int val=G[x][i].val;
		if(u==f)continue;
		//给U[u]一个初值  
		if(F1[x]==u)U[u]=D2[x]+val;//u是x向下最远路劲上的孩子
		else U[u]=D1[x]+val;
	}
	for(int i=0;i<G[x].size();i++){
		int u=G[x][i].x;
		int val=G[x][i].val;
		if(u==f)continue;
		U[u]=max(U[u],U[x]+val);//更新U[u] 
		work(u,x);
	}
}
int main(){
	int n,x,val;
	while(~scanf("%d",&n)){
		init(n);
		for(int i=2;i<=n;i++){
			scanf("%d%d",&x,&val);
			G[i].push_back(node(x,val));
			G[x].push_back(node(i,val));
		}
		dfs(1,0);
		work(1,0);
		for(int i=1;i<=n;i++)
		printf("%d\n",max(D1[i],U[i]));
	}
}

这题感觉又增长了见识，之前做过求树的直径的，才知道树的直径就是两次Bfs即可，这一题则是求树的任意结点到其他结点的最远距离。