【Week6作业A】氪金带东

题目描述

实验室里原先有一台电脑(编号为1),最近氪金带师咕咕东又为实验室购置了N-1台电脑,编号为2到N。每台电脑都用网线连接到一台先前安装的电脑上。但是咕咕东担心网速太慢,他希望知道第i台电脑到其他电脑的最大网线长度,但是可怜的咕咕东在不久前刚刚遭受了宇宙射线的降智打击,请你帮帮他。
在这里插入图片描述
提示: 样例输入对应这个图,从这个图中你可以看出,距离1号电脑最远的电脑是4号电脑,他们之间的距离是3。 4号电脑与5号电脑都是距离2号电脑最远的点,故其答案是2。5号电脑距离3号电脑最远,故对于3号电脑来说它的答案是3。同样的我们可以计算出4号电脑和5号电脑的答案是4.

输入格式

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

输出格式

对于每组测试数据输出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,找到的最远点必为端点之一,再从该端点出发进行DFS,同时记录其到各点的距离,并找到最远点,为另一端点,再从另一端点出发进行DFS,若该点到某一点的距离大于之前记录的距离,则对距离值进行更新,经过这三次遍历之后,我们所记录的距离,即为各点到最远点的距离。

代码

#include <iostream>
#include <vector>
#include<string.h> 
using namespace std;
const int size=1e4+10; 
vector<pair<int,int>> a[size];
int Max[size];
bool vis[size];
int far;
void dfs(int start,int length)
{
	vis[start]=true;
	for(int i=0;i<a[start].size();i++)
	{
		int temp=a[start][i].first;
		if(!vis[temp])
		{
			Max[temp]=max(Max[temp],length+a[start][i].second);
			far=Max[far]<Max[temp]?temp:far;
			dfs(a[start][i].first,length+a[start][i].second);
		}	
	}
}
int main(int argc, char** argv) {
	int n;
	while(~scanf("%d",&n))
	{
		for(int i=1;i<=n;i++)
			a[i].clear();
		memset(Max,0,sizeof(Max));
		memset(vis,0,sizeof(vis));
		for(int i=2;i<=n;i++)
		{
			int e1,e2;
			scanf("%d%d",&e1,&e2);
			a[i].push_back({e1,e2});
			a[e1].push_back({i,e2});
		}
		far=1;
		dfs(1,0);
		memset(vis,0,sizeof(vis));
		dfs(far,0);
		memset(vis,0,sizeof(vis));
		dfs(far,0);
		for(int i=1;i<=n;i++)
			printf("%d\n",Max[i]);
	}
	return 0;
}
发布了24 篇原创文章 · 获赞 8 · 访问量 523

猜你喜欢

转载自blog.csdn.net/weixin_44034698/article/details/105161101