CSP201503-4(树的直径+bfs)

2 篇文章 0 订阅
1 篇文章 0 订阅

在这里插入图片描述
求树的直径裸题。可以用两次dfs或者两次bfs,都一样。
树的直径:从任意一点到距离这一点的最远处的一点,最远处的这一点为树的直径的一个端点,从这个端点开始深搜或者广搜求得距离该端点最远的一个结点,也就是树的直径的另一个端点。

#include <iostream>
#include <queue>
#include <vector> 
using namespace std;
typedef struct{
	int p;
	int level;
	int pace=0;
	vector<int> graph;//储存边,如果用邻接矩阵表示法开最大值是0分哟 
}INTER;
vector<INTER> inter(1);
int main(){
	int n,m;
	cin>>n>>m;
	INTER node;
	node.level=0;
	node.p=1;
	inter.push_back(node);
	int maxl=0;
	for (int a=2;a<=n;a++){
		int t;
		scanf ("%d",&t);
		inter[t].graph.push_back(a);
		INTER temp;
		temp.level=inter[t].level+1;
		if (maxl<temp.level) maxl=temp.level;
		temp.p=a;
		temp.graph.push_back(t);
		inter.push_back(temp);
	}
	for (int a=n+1;a<=n+m;a++){//既然后面的表示不涉及是交换机还是电脑,干脆直接把电脑的编号根据交换机数目往后顺移,方便计算 
		int t;
		scanf ("%d",&t);
		inter[t].graph.push_back(a);
		INTER temp;
		temp.level=inter[t].level+1;
		if (maxl<temp.level) maxl=temp.level;
		temp.p=a;
		temp.graph.push_back(t);
		inter.push_back(temp);
	}
	queue<INTER> q;//用队列保存深度最大的结点,相当于从根开始找离根最远的结点,也即求树的直径的第一步,任一结点求距离其最远结点 
	for (auto it=inter.begin();it!=inter.end();it++){
		if ((*it).level==maxl) q.push(*it);
	}
	long long maxpace=0;//保存最大距离 
	while (!q.empty()){//对于每一个距离根节点最远的子结点 
		queue<INTER>que;//便于对值修改,开一个临时队列 
		que.push(q.front());
		vector<int> fl(n+m+1);
		while (!que.empty()){//bfs求最大距离,即树的直径 
			auto t=que.front();
			fl[t.p]=1;
			for (auto it=inter[t.p].graph.begin();it!=inter[t.p].graph.end();it++){
				if (!fl[(*it)]){
					inter[(*it)].pace=inter[t.p].pace+1;
					if (maxpace<inter[(*it)].pace) maxpace=inter[(*it)].pace;
					que.push(inter[(*it)]);
					fl[(*it)]=1;
				}
			}
			que.pop();
		}
		for (auto it=inter.begin();it!=inter.end();it++){
			(*it).pace=0;
		}
		q.pop();
	}
	printf ("%lld",maxpace);//是不是long long我也没试过,反正用了也不会怎么样 
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值