如果将森林里每棵树都各自看做一个点,那么最后所连成的树应该是一颗菊花,否则将叶子节点父亲改为根不会更劣。
对于每个点所代表的树,其和根节点相连的点应该是到其他点距离最大值最小的点。这个点显然是直径的中点。而距离最大值就可以作为点的权值了。
显然根节点应该是权值最大的。这样连完后讨论一下答案即可。
#include<iostream> #include<cstdio> #include<cmath> #include<cstdlib> #include<cstring> #include<algorithm> using namespace std; int read() { int x=0,f=1;char c=getchar(); while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();} while (c>='0'&&c<='9') x=(x<<1)+(x<<3)+(c^48),c=getchar(); return x*f; } #define N 500010 #define inf 1000000000 int n,m,l,p[N],deep[N],fa[N],value[N],cnt=0,t=0,ans; bool flag[N]; struct data{int to,nxt,len; }edge[N<<1]; void addedge(int x,int y,int z){t++;edge[t].to=y,edge[t].nxt=p[x],edge[t].len=z,p[x]=t;} int dfs1(int k) { int mx=k;flag[k]=1; for (int i=p[k];i;i=edge[i].nxt) if (!flag[edge[i].to]) { deep[edge[i].to]=deep[k]+edge[i].len; int x=dfs1(edge[i].to); if (deep[x]>deep[mx]) mx=x; } return mx; } int dfs2(int k) { int mx=k; for (int i=p[k];i;i=edge[i].nxt) if (edge[i].to!=fa[k]) { fa[edge[i].to]=k; deep[edge[i].to]=deep[k]+edge[i].len; int x=dfs2(edge[i].to); if (deep[x]>deep[mx]) mx=x; } return mx; } int main() { #ifndef ONLINE_JUDGE freopen("bzoj3246.in","r",stdin); freopen("bzoj3246.out","w",stdout); const char LL[]="%I64d\n"; #else const char LL[]="%lld\n"; #endif n=read(),m=read(),l=read(); for (int i=1;i<=m;i++) { int x=read()+1,y=read()+1,z=read(); addedge(x,y,z),addedge(y,x,z); } for (int i=1;i<=n;i++) if (!flag[i]) { deep[i]=0;int root=dfs1(i); deep[root]=0;int x=dfs2(root); value[++cnt]=inf;ans=max(ans,deep[x]); for (int y=x;y;y=fa[y]) value[cnt]=min(value[cnt],max(deep[x]-deep[y],deep[y])); } sort(value+1,value+cnt+1);reverse(value+1,value+cnt+1);value[cnt+1]=value[cnt+2]=-inf; cout<<max(ans,max(value[1]+value[2]+l,value[2]+value[3]+(l<<1))); return 0; }