【图论】求树的直径(最长链)

最新推荐文章于 2022-11-13 11:00:16 发布

易橙

最新推荐文章于 2022-11-13 11:00:16 发布

阅读量555

点赞数

CC 4.0 BY-SA版权

分类专栏：图论

本文链接：https://blog.youkuaiyun.com/Rainfoo/article/details/105290837

图论专栏收录该内容

37 篇文章

订阅专栏

本文介绍了利用深度优先搜索（DFS）的双指针方法来求解树的直径问题，即找到树中最长的路径。通过从任意节点P开始，找到最远节点Q，然后再从Q出发找到最远节点W，W与Q之间的距离即为树的直径。

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

方法有很多，我这里用2种dfs的做法来做
方法：先从任意一点P出发，找离它最远的点Q，再从点Q出发，找离它最远的点W，W到Q的距离就是是的直径

#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<stack>
using namespace std;
#define ll long long
const int maxn=2e5+5;
int d[maxn],head[maxn],f_num,ans,tot;
struct E{
	int to,next,w;
}edge[maxn];
void add(int u,int v,int z){
	edge[tot].to=v;
	edge[tot].w=z;
	edge[tot].next=head[u];
	head[u]=tot++;
}
void dfs(int x,int fa){
	if(ans<d[x]){
		ans=d[x];
		f_num=x;
	}
	for(int i=head[x];i!=-1;i=edge[i].next){
		int v=edge[i].to;
		if(v==fa) continue;
		d[v]=d[x]+edge[i].w;
		dfs(v,x);
	}
}
int main(){
	memset(head,-1,sizeof(head));
	int n,m;cin>>n>>m;
	for(int i=1;i<=m;i++){
		int u,v,w;cin>>u>>v;
		//cin>>w;
		add(u,v,w);add(v,u,w);
	}
	dfs(1,0);
	ans=0;
	d[f_num]=0;
	dfs(f_num,0);
	cout<<ans<<endl;
}