我们枚举切掉的那条边,分成了两个子树,分别dfs一遍求出直径,再dfs一遍求出中心(即距离最远点最近的点),那么把这两棵子树连起来之后的直径最小就是max(d1,d2,r1+r2+w)。
复杂度
O(n2)
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 5010
inline char gc(){
static char buf[1<<16],*S,*T;
if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
return *S++;
}
inline int read(){
int x=0,f=1;char ch=gc();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
return x*f;
}
int n,h[N],num=1,mx1[N],mx1id[N],mx2[N],mx2id[N],ans=inf,res,r;
struct edge{
int to,next,val;
}data[N<<1];
inline void dfs1(int x,int Fa){//求直径
for(int i=h[x];i;i=data[i].next){
int y=data[i].to;if(y==Fa) continue;
dfs1(y,x);int w=mx1[y]+data[i].val;
if(w>mx1[x]) mx2[x]=mx1[x],mx2id[x]=mx1id[x],mx1[x]=w,mx1id[x]=y;
else if(w>mx2[x]) mx2[x]=w,mx2id[x]=y;
}res=max(res,mx1[x]+mx2[x]);
}
inline void dfs2(int x,int Fa){//求中心
r=min(r,mx1[x]);
for(int i=h[x];i;i=data[i].next){
int y=data[i].to;if(y==Fa) continue;
int w=0;if(mx1id[x]==y) w=mx2[x]+data[i].val;
else w=mx1[x]+data[i].val;
if(w>mx1[y]) mx2[y]=mx1[y],mx2id[y]=mx1id[y],mx1[y]=w,mx1id[y]=x;
else if(w>mx2[y]) mx2[y]=w,mx2id[y]=x;dfs2(y,x);
}
}
int main(){
// freopen("a.in","r",stdin);
n=read();
for(int i=1;i<n;++i){
int x=read(),y=read(),val=read();
data[++num].to=y;data[num].next=h[x];h[x]=num;data[num].val=val;
data[++num].to=x;data[num].next=h[y];h[y]=num;data[num].val=val;
}for(int i=1;i<n;++i){
int x=data[i<<1].to,y=data[i<<1|1].to;memset(mx1,0,sizeof(mx1));
memset(mx2,0,sizeof(mx2));res=0;dfs1(x,y);dfs1(y,x);r=inf;dfs2(x,y);
int r1=r;r=inf;dfs2(y,x);res=max(res,r1+r+data[i<<1].val);
ans=min(ans,res);
}printf("%d\n",ans);
return 0;
}