week 6 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,但这道题是求从树的每个点出发的最长链长度,则要比较任意节点到组成此树直径的两端点的距离大小,即三次dfs。
任选一个点0, 从0深搜找到距它最远的点x,求树上每个点到x的距离d1[],再从x开始深搜找到距x最远的点,将其再赋给x,接下来求树上每个点到x的距离d2[], 那么第i个点出发的最长链长度为max(d1[i], d2[i]).

代码:

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

const int maxn = 10010;
int head[maxn], d1[maxn], d2[maxn], x, n, tot;
struct edge 
{ 
	int to, next, w;
}edges[2 * maxn];

void add(int u, int v, int w) 
{
	edges[tot] = { v, head[u], w };
	head[u] = tot;
	tot++;
}

void dfs(int u, int f, int* d) 
{
	for (int e = head[u]; e != -1; e = edges[e].next) 
	{
		int to = edges[e].to;
		if (to != f)
		{
			d[to] = d[u] + edges[e].w;
			if (d[x] < d[to])
				x = to;
			dfs(to, u, d);
		}
	}
}

int main() 
{
	int v, w;
	while (scanf_s("%d", &n)!=EOF) 
	{
		memset(head, -1, sizeof(head));
		tot = 0;
		for (int i = 2; i <= n; i++) 
		{
			scanf_s("%d%d", &v, &w);
			add(i, v, w);
			add(v, i, w);
		}
		d1[x] = 0;
		dfs(1, -1, d1);
		d1[x] = 0;
		dfs(x, -1, d1);
		d2[x] = 0;
		dfs(x, -1, d2);
		for (int i = 1; i <= n; i++)
			printf("%d\n", max(d1[i], d2[i]));
	}
	return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值