求树的直径(BFS/DFS)

最新推荐文章于 2024-10-24 17:14:16 发布

原创

最新推荐文章于 2024-10-24 17:14:16 发布 · 1.3k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#树 #DFS/BFS

树的直径概念：一颗树上存在的最长路径。

推导过程请参考：https://blog.youkuaiyun.com/enjoying_science/article/details/44962389

代码如下(BFS版)：

#include<iostream>
#include<stdio.h>
#include<map>
#include<vector>
#include<set>
#include<cstdlib>
#include<string.h>
#include<string>
#include<queue>
#include<algorithm>
#include<cmath>
#define MAXN 5000000
#define LL long long
#define EPS 1e-9
using namespace std;
struct Edge{
	int to;
	int next;
	int w;
}e[MAXN];
int tot,n,max_num,u,v,mst,med;
int dis[MAXN];
bool vis[MAXN];
int head[MAXN];
void init()
{
	tot=0;
	for(int i=1;i<=n;i++)
	{
		head[i]=-1;
	}
}
void add(int from,int to,int w)
{
	e[tot].to=to;
	e[tot].w=w;
	e[tot].next=head[from];
	head[from]=tot++;
}
int maxd;
void bfs(int st)
{
	queue<int>q;
	int i,now,next;
	memset(vis,false,sizeof(vis));
	memset(dis,0,sizeof(dis));
	vis