week6作业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进行遍历。第一次遍历从任意一点开始搜索距离该点最远的节点v1,第二次遍历,从v1点开始搜索距离该点最远的节点v2,v1与v2间的距离就是树的直径。第三次遍历从v2点开始搜索,因为在遍历中,第二次遍历将每个点与v1点的距离进行存储,第三次遍历将每个点与v2点的距离进行存储,通过比较大小,将距离每个点与v1,v2的距离中将较远的距离存储到数组中,求出第i台电脑与其他电脑距离最大的网线长度并将其输出。

代码:

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
struct node{
	int begin;
	int end;
	int value;
	int next;
}temp[100050];
int cm=1;
int a[100050];
int n;
int vis[100050];
int bm[100050];
int dm[100050];
void init(int &n)
{
	//cm=1;
	cm=0;
	for(int i=0;i<=n;i++)
	{
		a[i]=-1;
		vis[i]=0;
		bm[i]=0;
		dm[i]=0;
	}
 } 
void edge(int begin,int end,int value)
{
	temp[cm].begin=begin;
	temp[cm].end=end;
	temp[cm].value=value;
	temp[cm].next=a[begin];
	a[begin]=cm;
	cm++;
}

int v1=0;
int v2=0;
void dfs1(int begin)
{
	vis[begin]=1;
	for(int i=a[begin];i!=-1;i=temp[i].next)
	{
		if(bm[temp[i].end]==-1)
		//if(!vis[temp[i].end])
		{
			vis[temp[i].end]=1;
			bm[temp[i].end]=bm[temp[i].begin]+temp[i].value;
			if(bm[temp[i].end]>bm[v1])
			{
				v1=temp[i].end;
			}
			dfs1(temp[i].end);
		}
	}
}
void dfs2(int begin)
{
	vis[begin]=1;
	for(int i=a[begin];i!=-1;i=temp[i].next)
	{
		if(bm[temp[i].end]==-1)
		//if(!vis[temp[i].end])
		{
			vis[temp[i].end]=1;
			bm[temp[i].end]=bm[temp[i].begin]+temp[i].value;
			if(bm[temp[i].end]>bm[v2])
			{
				v2=temp[i].end;
			}
			dfs2(temp[i].end);
		}
	 } 
}
void dfs3(int begin)
{
	vis[begin]=1;
	for(int i=a[begin];i!=-1;i=temp[i].next)
	{
		if(dm[temp[i].end]==-1)
		//if(!vis[temp[i].end])
		{
			vis[temp[i].end]=1;
			dm[temp[i].end]=dm[temp[i].begin]+temp[i].value;
			if(dm[temp[i].end]>bm[temp[i].end])
			{
				bm[temp[i].end]=dm[temp[i].end];
			}
			dfs3(temp[i].end);
		}
	 } 
}
int main()
{
	int end;
	int value;
	while(cin>>n)
	{
		init(n);
		for(int i=2;i<=n;i++)
		{
			cin>>end>>value;
			edge(end,i,value);
			edge(i,end,value);
		}
		for(int i=0;i<=n;i++)
		{
			bm[i]=-1;
		}
		bm[1]=0;
		dfs1(1);
		for(int i=0;i<=n;i++)
		{
			bm[i]=-1;
			//bm[i]=0;
		}
		bm[v1]=0;
		dfs2(v1);
		for(int i=0;i<=n;i++)
		{
			dm[i]=-1;
			//dm[i]=0;
		}
		dm[v2]=0;
		dfs3(v2);
                for(int i=1;i<=n;i++)
		{	cout<<bm[i]<<endl;
		}
	}
	return 0;
}
发布了19 篇原创文章 · 获赞 0 · 访问量 204

猜你喜欢

转载自blog.csdn.net/weixin_45117273/article/details/105333245