WEEK6 周记 作业——图与树_氪金带东

WEEK6 周记 作业——图与树_氪金带东

一、题意

1.简述

实验室里原先有一台电脑(编号为1)，最近氪金带师咕咕东又为实验室购置了N-1台电脑，编号为2到N。每台电脑都用网线连接到一台先前安装的电脑上。希望知道第i台电脑到其他电脑的最大网线长度。

2.输入格式

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

3.输出格式

对于每组测试数据输出N行，第i行表示i号电脑的答案 (1<=i<=N).

4.样例

Input

5
1 1
2 1
3 1
1 1

Output

3
2
3
4
4

样例解释

样例输入对应这个图，从这个图中你可以看出，距离1号电脑最远的电脑是4号电脑，他们之间的距离是3。 4号电脑与5号电脑都是距离2号电脑最远的点，故其答案是2。5号电脑距离3号电脑最远，故对于3号电脑来说它的答案是3。同样的我们可以计算出4号电脑和5号电脑的答案是4.

二、算法

主要思路

首先给出一个定理：树的直径一定是某两个叶子之间的距离，从树中任选一个点开始遍历这棵树，找到一个距离这个点最远的叶子，然后再从这个叶子开始遍历，找到离这个叶子最远的另一个叶子，他俩之间的距离就是树的直径。
个人的证明（理解）：反证法。
假设从任意一点（这里称为$A$点）开始遍历找到距离A点最远的点（$B$点）不是直径的端点（也即$B$点与最远点$C$点的距离不是最大的），那么由于树是连通的，从A点开始必然能够遍历到这个树的直径的某一个端点。设直径的两个端点为$D_1$和$D_2$，由于这是一棵树，所以从$D_1$到$D_2$必然只有一条无重复点的路径，而$A$要想遍历到$D_1$或$D_2$，必然先遍历到这条路径中间的某个点（$M$点），再沿这条路径到达端点，否则就出现了回路。从$A$到$B$的路径也必然经过一部分从$D_1$到$D_2$路径的点（其中包括$M$），否则$B$与$D_2$的距离要比$D_1$与$D_2$的距离长。设$MN$为重合的路径部分，$B$点要想到达$D_2$必然走$BN+N{D}_2$（否则就有两条路径了，会出现回路），而$D_1{D}_2=D_{1}N+N{D}_2$，其中$BN$要比$D_{1}N$长，这就与$D_1{D}_2$是直径矛盾。

---

知道这个定理之后，这个题就很好做了。
先随便找一个点（A点），然后从这个点开始dfs，一边遍历一边记录下来与A点的距离，遍历完之后找出距离最大的点（$D_1$），然后从这个点开始再进行一次dfs，记录下$D_1$与所有点的距离，找出与$D_1$距离最大的点$D_2$，再用$D_2$进行一次dfs，得到$D_2$与所有点的距离。
根据定理，任意一个点的最远点必然是直径的端点，也即$D_1$和$D_2$两个点中的一个，那么对于任意一个点，我们都知道了其与两个端点的距离，我们直接去这两个距离中的max就行了。

三次dfs解决！

---

题外

本题当时出错的地方：忘记多组数据了= =
给我的debug提了一个醒：题意的复查很重要。

三、代码

#include<iostream>
#include<vector>
#include<cstring>
#include<algorithm>
using namespace std;
//3次dfs,找到树的直径的端点
long long int dis[10010];
int vis[10010];//从1开始 
long long int dv1[10010];
long long int dv2[10010];
vector<pair<int,int> > v[10010];//从1开始
int n;

void dfs(int cmpu,long long int dis[],long long int d){
	for(int i=0;i<v[cmpu].size();i++){//size时间复杂度是O（1） 
		if(vis[v[cmpu][i].first]==1) continue;
		vis[v[cmpu][i].first] = 1;
		dis[v[cmpu][i].first] = d+v[cmpu][i].second;
		dfs(v[cmpu][i].first,dis,d+v[cmpu][i].second);
	}
}
int main()
{
	while(scanf("%d",&n)!=EOF)
	{
		memset(dis,0,sizeof(dis));
		memset(vis,0,sizeof(vis));
		memset(dv1,0,sizeof(dv1));
		memset(dv2,0,sizeof(dv2));
		memset(v,0,sizeof(v));
		int i;
		for(i=2;i<=n;i++){
			int ad,lth;
			scanf("%d%d",&ad,&lth);
			v[i].push_back({ad,lth});
			v[ad].push_back({i,lth});
		}
		//默认1号是第一个安装好的 
		vis[1] = 1;
		dis[1] = 0;
		dfs(1,dis,0);
		int max = 1;
		for(i=1;i<=n;i++)
			if(dis[i]>dis[max]) 
				max = i;
	
		memset(vis,0,sizeof(vis));
		vis[max] = 1;
		dv1[max] = 0;
		dfs(max,dv1,0);
		for(i=1;i<=n;i++)
			if(dv1[i]>dv1[max]) 
				max = i;
		
		memset(vis,0,sizeof(vis));
		dv2[max] = 0;
		vis[max] = 1;
		dfs(max,dv2,0);
		
		for(i=1;i<=n;i++){
			long long int maxv;
			maxv = dv1[i]>dv2[i]?dv1[i]:dv2[i];
			printf("%lld\n",maxv);
		}
	}
	return 0;
}