图的遍历 树的直径

题意:

实验室里原先有一台电脑(编号为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


思路:

这道题是为了每个点距离这个树中最远的顶点的距离,粗略思考可以对每个点遍历一遍得到每个点距离这个点的距离然后取最大值,但是这样做太过暴力超时,因此这时我们可以利用树的直径,在树中,距离某个点最远的点一定是树的直径的端点,所以我们只要求得树的直径的端点,然后求得树中的每个点距离这两个端点的最大值,这里有个很巧妙的方法,我们得到树的直径两端点后,可以不从树中点出发,而是从这两个端点出发进行遍历,采用bfs的方法,两次遍历便可以得到树中每个点分别距离这两个端点的距离,去两者的最大值,便可以得到最远距离。

如何求树的直径:
假设树的最长路的两个叶子节点为v1,v2,从任意一点u出发走到的最 远的点一定是(v1,v2)中的一点,然后再从v1或者v2出发走到的最 远点一定是v2或者v1。由此经过两次遍历就能找到最长路径

注意我们这里存图的方式,采用邻接表的方式,不能采用邻接矩阵的方式,不然会超内存。


总结:

当我们想要求得树中某点距离这个树中最远的顶点的距离时我们可以利用树的直径。

求树的直径的方法:
假设树的最长路的两个叶子节点为v1,v2,从任意一点u出发走到的最 远的点一定是(v1,v2)中的一点,然后再从v1或者v2出发走到的最 远点一定是v2或者v1。由此经过两次遍历就能找到最长路径


代码:

#include<stdio.h>
#include<string.h>
#include<queue>
#include<vector>
using namespace std;

struct Edge
{
	int v;
	int w;
	
	Edge(int _v = 0, int _w = 0) :v(_v), w(_w) {}
	Edge(const Edge& t) :v(t.v), w(t.w) {}
};

vector<Edge>computer[10001];
long long length1[10001];

bool arrived[10001];
queue<int>q;
int n;
long long length[10001];

void bfs(long long* leng)
{
	while (!q.empty())
	{
		int now = q.front();
		q.pop();
		int size = computer[now].size();
		for (int i = 0; i < size; i++)
		{
			if (arrived[computer[now][i].v] == false)
			{
				q.push(computer[now][i].v);
				leng[computer[now][i].v] = leng[now] + computer[now][i].w;
				arrived[computer[now][i].v] = true;
			}
		}
	}
}

int main()
{
	while (scanf_s("%d", &n)!=EOF)
	{
	    
		for (int i = 0; i < 10001; i++)computer[i].clear();
		memset(arrived, false, sizeof(bool) * 10001);
		int com;
		long long len;
		for (int i = 2; i <= n; i++)
		{
			scanf_s("%d %lld", &com, &len);
			computer[i].push_back(Edge(com,len));
			computer[com].push_back(Edge(i,len));
		}


		length[1] = 0;
		int lo;
		arrived[1] = true;

		q.push(1);
		bfs(length);

		int which = 0;
		long long max = -1;
		for (int i = 1; i <= n; i++)
		{
			if (max < length[i])
			{
				max = length[i];
				which = i;
			}
		}

		int oneside = which;

		memset(length, 0, sizeof(long long) * 10001);
		memset(arrived, false, sizeof(bool) * 10001);
		arrived[oneside] = true;
		q.push(oneside);
		bfs(length);

		which = 0;
		max = -1;
		for (int i = 1; i <= n; i++)
		{
			if (max < length[i])
			{
				max = length[i];
				which = i;
			}
		}
		int otherside = which;

		memset(arrived, false, sizeof(bool) * 10001);
		arrived[otherside] = true;
		q.push(otherside);
		length1[otherside] = 0;
		bfs(length1);

		for (int i = 1; i <= n; i++)
		{
			long long ans;
			if (length[i] > length1[i])ans = length[i];
			else ans = length1[i];
			printf("%lld\n", ans);
		}
	}
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值