程序设计思维与实践 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).

Example

Input
5
1 1
2 1
3 1
1 1
Ouput
3
2
3
4
4

树的直径总结

做法
  • 首先明确一点:树的直径一定是某两个叶子之间的距离
  • 从树中任选一个点开始遍历这棵树,找到一个距离这个点最远的叶子,然后再从这个叶子开始遍历,找到离这个叶子最远的另一个叶子,它们之间的距离就是树的直径
  • 两次遍历即可求得树的直径
  • 遍历可以使用DFS也可以使用BFS,找到距离起点最远的叶子结点就好
模板(借鉴)

树的直径求法模板

思想

  • 此题要找的不是最长路径,是找某一个节点x所能到达的最长路径
  • 对于树中的一个节点,它的最长路径是向右延展的最长路径向左延展的最长路径最大值
  • 由于不知道往哪边走是最长路径,所以在找到树的直径的情况下,要分别从右边的最远点和左边的最远点再次遍历,以找到直径的两个端点到每一个节点的距离。

代码

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;

const int maxn=11000;
struct node
{ 
	int from,end,next,val; 
}edge[2*maxn];
int head[maxn];
int vis[maxn];//标记是否到达过 
int dis1[maxn],dis2[maxn];//左右距离 
int maxv;//起点能到达的最远点 
int edgenum;//边数 

void add(int u,int v,int w) 
{
	edge[edgenum].from=u;
	edge[edgenum].end=v;
	edge[edgenum].next=head[u];
	edge[edgenum].val=w;
	head[u]=edgenum++;
}

void dfs(int last,int f,int* d) 
{
	for (int i=head[last];i!=-1;i=edge[i].next) 
	{
		int v=edge[i].end;
		int w=edge[i].val;
		if(v!=f)
		{
			d[v]=d[last]+edge[i].val;
	        if (d[maxv]<d[v]) maxv=v;
			dfs(v,last,d);
	    }
	}
}

int main() {
	int n,a,value,begin;
	while (~scanf("%d",&n)) 
	{
		memset(head,-1,sizeof(head));
		edgenum=0;
		for (int i=2;i<=n;i++) 
		{
			scanf("%d%d",&a,&value);
			add(i,a,value);
			add(a,i,value);//加双向边 
		}
		dis1[1]=0;
		begin=1,
		dfs(1,0,dis1);//求直径 
		begin=maxv;
		dis1[maxv]=0;
		dfs(maxv,0,dis1);//求一边最远距离 
		begin=maxv;
		dis2[maxv]=0;
		dfs(maxv,0,dis2);//求另一边最远距离 
		for (int i=1;i<=n;i++)
			printf("%d\n", max(dis1[i],dis2[i]));
	}
	return 0;
}
发布了7 篇原创文章 · 获赞 0 · 访问量 296

猜你喜欢

转载自blog.csdn.net/qq_45337415/article/details/105117373