week6作业题_A-氪金带东

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

题目思路

根据题意可知，该题是一个求每个节点到其他节点最长路的问题。在这里就涉及到一个概念，即树的直径。树的直径就是树中任意两点之间的距离的最大值。
求树的直径肯定是两个叶子之间的距离。我们可以从数中任选一点开始遍历这棵树，找到一个距离这个点最远的叶子，然后再从这个叶子开始遍历，找到离这个叶子最远的另一个叶子，他俩之间的距离就是树的直径。即只需要两次dfs或者bfs遍历就可以求得树的直径。
在我们这个题目中，树中任意一点到树中最远的点一定是直径的两个叶节点*（v1，v2）中的一个。所以我们就可以使用三次遍历来实现这个过程，首先从任意一个节点开始，遍历一次找到直径的一个节点v1*，然后从这个节点再一次遍历，并且记录各个节点到v1的距离，并且找到直径的另一个节点v2，然后从这个节点v2遍历一次，记录每个节点到v2的距离。然后在这两个距离之间求个max就可以了。

代码实现

//#include <bits/stdc++.h>
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define _for(i,a,b) for(int i = (a); i < (b); i++)
#define _rep(i,a,b) for(int i = (a); i <= (b); i++)
using namespace std;
const int MAXN = 1e5 + 10;

//链式前向星 
struct Edge{
	int v,w,next;	//v是邻接点，w是边的权值，next是该点的下一个邻接点的边edge的下标 
}Edges[MAXN];

int n,a,b,ans; 
int head[MAXN],tot;	//tot记录Edges的下标
bool vis[MAXN];
int dis1[MAXN] ,dis2[MAXN];
int v1;			//直径的一个端点 

void addEdge(int u,int v,int w)		//添加一条边 
{
	Edges[tot].v = v;
	Edges[tot].w = w;
	Edges[tot].next = head[u];
	head[u] = tot;
	tot++;
}

void dfs(int v)//dis记录从起始点到各点的距离 
{
	vis[v] = true;
	if(dis1[v] > ans)
	{
		ans = dis1[v];
		v1 = v; 
	}
	for(int i = head[v]; i > 0; i = Edges[i].next)
	{
		int u = Edges[i].v;
		if(vis[u] == false)
		{
			dis1[u] = dis1[v] + Edges[i].w;
			dfs(u); 
		}
	}
}

void read()
{
	ios::sync_with_stdio(false);		//关闭同步
	tot = 1;
	_rep(i,2,n)
	{
		cin >> a >> b;
		addEdge(i,a,b);
		addEdge(a,i,b);
	}
	
} 

int main()
{
	while(cin >> n)
	{
		memset(head,-1,sizeof(head));
		read();
		memset(dis1,0,sizeof(dis1));
		memset(dis2,0,sizeof(dis2));
		memset(vis,false,sizeof(vis));
		ans = 0;
		v1 = 1;
		dfs(1);
		memset(dis1,0,sizeof(dis1));
		memset(vis,false,sizeof(vis));

		ans = 0;
		dfs(v1);
		_rep(i,1,n)
			dis2[i] = dis1[i];
		memset(dis1,0,sizeof(dis1));
		memset(vis,false,sizeof(vis)); 
		ans = 0;
		dfs(v1);
		_rep(i,1,n)
		{
			cout << max(dis1[i],dis2[i]) << endl;
		}
	}
}