树上直径求法
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;
}