poj 2631 (求树上的最大距离基础题)

最新推荐文章于 2020-12-03 22:08:37 发布

原创最新推荐文章于 2020-12-03 22:08:37 发布 · 752 阅读

0 ·

CC 4.0 BY-SA版权

Poj Solutions 同时被 2 个专栏收录

13 篇文章

订阅专栏

树的题目

3 篇文章

订阅专栏

该博客介绍了如何解决POJ 2631问题，即在树形结构中找到最大距离。通过两次广度优先搜索（BFS），首先从任意点出发找到最远节点，然后从该最远点再次进行BFS找到新的最远点，这两个点之间的距离即为树的最大路径长度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：http://poj.org/problem?id=2631

求树上的最大距离

方法：任选一点bfs，求出其他点到此点的最短距离，然后选取距离最大的点（此点为最长路的一个端点）再做一次bfs，然后选取距离最大的点（此点为最长路的另外一个端点）二者之间的距离即为树上的最大距离。

code:

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <cstring>
#define INF 1000000000

using namespace std;

const int maxn=10050;
const int maxe=40000;

struct edge
{
	int to,next,cost;
} P[maxe];
int head[maxn],si;
int dis[maxn],que[maxe],nn;

void add_edge(int s,int t,int cc)
{
	P[si].to=t;
	P[si].cost=cc;
	P[si].next=head[s];
	head[s]=si++;
}

void bfs(int u)
{
	for(int i=0;i<=nn;i++) dis[i]=INF;
	int hh,tt;
	hh=tt=0;
	que[tt++]=u;
	dis[u]=0;
	while(hh!=tt){
		int v=que[hh++];
		for(int i=head[v];i!=-1;i=P[i].next){
			int mid=P[i].to;
			if(dis[v]+P[i].cost<dis[mid]){
				dis[mid]=dis[v]+P[i].cost;
				que[tt++]=mid;
			}
		}
	}
}
int main()
{
	int s,t,cc;
	nn=0;
	memset(head,-1,sizeof(head));
	si=0;
	while(scanf("%d%d%d",&s,&t,&cc)!=EOF){
		add_edge(s,t,cc);
		add_edge(t,s,cc);
		nn=max(nn,s);
		nn=max(nn,t);
	}
	bfs(1);
	s=1;
	for(int i=2;i<=nn;i++) if(dis[i]>dis[s]) s=i;
	bfs(s);
	t=1;
	for(int i=2;i<=nn;i++) if(dis[i]>dis[t]) t=i;
	printf("%d\n",dis[t]);
	return 0;
}