题意:给你N个点,N-1条边,组成一棵树,问从这棵树中任意取出一条边重新连接,并且使得连接后的还是一棵树(保持联通),要使最后重新组成的树的直径最小的那棵树的直径。
树的直径:树中任意两点之间距离最大的距离就是树的直径。
树的重心:树中任意一点到树中其他点最远距离最小的那个点。
树的重心一定在树的直径上。
思路:
1.先算出原树的直径上的点(计算树的直径直接bfs,同时记录路径)
2.枚举每条在直径上 的边,去掉后,分别计算两棵子树的直径,和重心到当前数中其他点的最远距离,得到m1,m2,和cg1,cg2,另记当前枚举的边的权值为w.
3.重组树的直径也就等于max(max(m1,m2),w+cg1+cg2);更新ans;
#include <iostream>
#include <cstdio>
#include <cstring>
#include <map>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn=2505;
const int maxm=5010;
int head[maxn],to[maxm],next[maxm],w[maxm],edge,maxl;
int n,dis[maxn],que[maxn],start,end;
bool vis[maxn];
struct node
{
int from,w;
} pre[maxn],path[maxn];
void init()
{
memset(head,-1,sizeof(head));
edge=0;
}
inline void add(int u,int v,int d)
{
to[edge]=v,w[edge]=d,next[edge]=head[u],head[u]=edge++;
to[edge]=u,w[edge]=d,next[edge]=head[v],head[v]=edge++;
}
void bfs(int s,int p)
{
memset(vis,0,sizeof(vis));
memset(dis,-1,sizeof(dis));
int l,r,v,u;
l=r=0;
vis[s]=1;
dis[s]=0;
que[r++]=s;
while(r>l)
{
u=que[l++];
for(int i=head[u]; ~i; i=next[i])
if(!vis[v=to[i]]&&to[i]!=p)
{
vis[v]=1;
dis[v]=dis[u]+w[i];
pre[v].from=u;
pre[v].w=w[i];
que[r++]=v;
}
}
}
void treediameter(int s,int p)
{
int u;
bfs(s,p);
maxl=0,u=s;
for(int i=0; i<n; i++)
if(dis[i]>maxl)
u=i,maxl=dis[i];
bfs(u,p);
int v=u;
maxl=0;
for(int i=0; i<n; i++)
if(dis[i]>maxl)
maxl=dis[i],v=i;
start=u,end=v;
}
int findCenter()
{
int res=end,tmp=dis[end];
int u=end;
while(true)
{
if(max(dis[end]-dis[u],dis[u]-dis[start])<tmp)
tmp=max(dis[end]-dis[u],dis[u]-dis[start]),res=u;
if(u==start) break;
u=pre[u].from;
}
return tmp;
}
int main()
{
int ca,u,v,d;
scanf("%d",&ca);
for(int t=1; t<=ca; t++)
{
scanf("%d",&n);
init();
for(int i=0; i<n-1; i++)
{
scanf("%d%d%d",&u,&v,&d);
add(u,v,d);
}
treediameter(0,-1);
u=end,v=start;
while(end!=start)
{
path[end].from=pre[end].from;
path[end].w=pre[end].w;
end=path[end].from;
}
int ans=1000000000;
while(u!=v)
{
int s=u,t=pre[u].from;
treediameter(s,t);
int m1=maxl;
int cg1=findCenter();
treediameter(t,s);
int m2=maxl;
int cg2=findCenter();
int tmp=path[u].w+cg1+cg2;
tmp=max(max(m1,m2),tmp);
ans=min(ans,tmp);
u=path[u].from;
}
printf("Case %d: %d\n",t,ans);
}
return 0;
}