树的直径

树上直径求法

void dfs(int u,int fa,int s)
{
    if(s>maxm){
        maxm=s; st=u;
    }
    for(int i=head[u];i;i=e[i].nxt){
        int v=e[i].to;
        if(v!=fa){
            dfs(v,u,s+1);
        }
    }
}

dfs(1,0,0)之后dfs(st,0,0)
dfs的中途可以维护关系
第一遍dfs会求出一条直径的边界点 第二遍dfs会求出直径

eg:
https://www.luogu.com.cn/problem/P4408
这道题是在一棵树上找到3个点A、B、C令AB+BC最大，同时要满足AB>AC。
证明不会 但可以去猜结论（）
找直径，直径端点为A,B,枚举C点则答案为max(min(dis[A][k],dis[B][k])+dis[A][B])
code:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define endl '\n'
struct edge{
	ll nxt,to,w;
}e[400050];
int head[200050],n,m,l,vis[200050];
ll dis[200050],dis1[200050],dis2[200050];
int cnt;
ll maxm=0,pos1,pos2;
ll ans=0;
void add(int u,int v,int w)
{
	e[++cnt].w=w;
	e[cnt].nxt=head[u];
	e[cnt].to=v;
	head[u]=cnt;
}
void dfs(int u,int fa)
{
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;
		if(v!=fa){
			dis[v]=dis[u]+e[i].w;
			dfs(v,u);
		}
	}
}
void dfs1(int u,int fa)
{
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;
		if(v!=fa){
			dis1[v]=dis1[u]+e[i].w;
			dfs1(v,u);
		}
	}
}
void dfs2(int u,int fa)
{
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;
		if(v!=fa){
			dis2[v]=dis2[u]+e[i].w;
			dfs2(v,u);
		}
	}
}
int main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
cin>>n>>m;
for(int i=1;i<=m;i++){
	ll x,y,z; cin>>x>>y>>z;
	add(x,y,z); add(y,x,z);
}
dfs(1,0);
for(int i=1;i<=n;i++){
	if(dis[i]>maxm){
		maxm=dis[i]; pos1=i;
		// cout<<maxm<<" "<<pos1<<endl;
	}
	dis[i]=0;
}
dfs(pos1,0);
for(int i=1;i<=n;i++){
	if(dis[i]>ans){
		ans=dis[i]; pos2=i;
	}
}
dfs1(pos1,0); dfs2(pos2,0);
ll anss=0;
for(int i=1;i<=n;i++){
	anss=max(anss,min(dis1[i],dis2[i]));
}
// cout<<ans<<" "<<anss<<endl;
cout<<anss+ans<<endl;
	return 0;
}