Computer(树的直径)

一所学校不久前买了第一台电脑(所以这台电脑的ID是1)。近年来,学校购买了N-1新电脑。每台新电脑都连接到一台先前安装的电脑上。学校的管理人员担心网络运行缓慢,希望知道第i台计算机需要发送信号的最大距离si(即到最远计算机的电缆长度)。您需要提供此信息。
在这里插入图片描述
提示:示例输入与此图对应。从图中,你可以看到计算机4离1最远,所以s1=3。计算机4和5是距离2最远的,所以s2=2。计算机5是离3最远的,所以s3=3。我们也得到了s4=4,s5=4。
输入
输入文件包含多组测试样例。在每组样例中,第一行中都有自然数n(n<=10000),然后是(n-1)行,其中包含对计算机的描述。第i行包含两个自然数-第i计算机所连接的计算机和用于连接的电缆长度。电缆总长度不超过10^9。输入行中的数字用空格分隔。
输出
对于每组样例,输出n行。第i行第i台计算机的到其他计算机的最大长度Si(1<=i<=n)。
样例输入
5
1 1
2 1
3 1
1 1
样例输出
3
2
3
4
4
提示
示例输入与此图对应。从图中,你可以看到计算机4离1最远,所以s1=3。计算机4和5是距离2最远的,所以s2=2。计算机5是离3最远的,所以s3=3。我们也得到了s4=4,s5=4。

思路:一台计算机到其他计算机最大的距离一定是树的直径对应的那两个点,所以就可以用直径对应的两个点然后bfs,保存距离,然后取到两个点距离的最大值。

ac代码:

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<queue>
#include<vector>
using namespace std;
const int N=40050;
typedef pair<int ,int> pii;
vector <pii > vt[N];
int ans;
int vis[N];
int dis[N];
int bfs(int x)
{
	int qp=0;
	memset(vis,0,sizeof(vis));
	memset(dis,0,sizeof(dis));
	ans=0;vis[x]=1;
	queue<int> q;
	q.push(x);
	while(!q.empty()){
		int t=q.front();
		q.pop();
		if(dis[t]>ans){
			ans=dis[t];
			qp=t;
		}
		for(int i=0;i<vt[t].size();i++){
			if(!vis[vt[t][i].first]){
				dis[vt[t][i].first]=dis[t]+vt[t][i].second;
				vis[vt[t][i].first]=1;
				q.push(vt[t][i].first);
			}
		}
	}
	return qp;
}
int main()
{
	int a,b,c;
	int n;
	while(cin>>n){
		for(int i=0;i<=N;i++)vt[i].clear();
	for(int i=2;i<=n;i++)
	{
		scanf("%d%d",&a,&b);
		vt[i].push_back(make_pair(a,b));
		vt[a].push_back(make_pair(i,b));
	}
	int qq=bfs(bfs(1));
	int ss[N];
	for(int i=1;i<=n;i++){
		ss[i]=dis[i];
	}
	bfs(qq);
	for(int i=1;i<=n;i++){
		int ansss=max(ss[i],dis[i]);
		cout<<ansss<<endl;
	}
	}
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值