ccf csp-201503-4-网络延时(树的直径+两次bfs)

树的直径定义:

一棵树中距离最远的两个点之间的距离就是树的直径。

树的直径算法:
1、首先以任意一个点为起点,通过dfs或bfs求出源点到所有点的路径长度。
2、从中选择长度最长的路径到达的点。
3、再次以这个点为源点重复上述步骤,得到的第二个最长的路径连接的两个点就是树的直径。

代码如下

#include<iostream>
#include<queue>
#include<cstring>
using namespace std;
const int N = 20005;
int n,m,len[N];
bool vis[N];
vector<int> v[N];
void bfs(){
	queue<int> q;
	q.push(1);
	int cur;
	while(!q.empty()){
		cur = q.front();
		q.pop();
		for(int i = 0; i < v[cur].size(); i++){
			int x = v[cur][i];
			if(vis[x]) continue;
			vis[x] = true;
			q.push(x);
		}
	}
	q.push(cur);
	memset(vis, false, sizeof(vis));
	while(!q.empty()){
		cur = q.front();
		q.pop();
		for(int i = 0; i < v[cur].size(); i++){
			int x = v[cur][i];
			if(vis[x]) continue;
			vis[x] = true;
			len[x] = len[cur]+1;
			q.push(x);
		}
	}
	printf("%d\n", len[cur]);
}
int main()
{
	int x;
	scanf("%d%d", &n, &m);
	for(int i = 2; i <= n+m; i++){
		scanf("%d", &x);
		v[i].push_back(x);
		v[x].push_back(i);
	}
	bfs();
	return 0;
}
发布了120 篇原创文章 · 获赞 758 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/qq_42437577/article/details/104693201
今日推荐