题意,移动一条边,使得树的直径最小。
首先枚举拆掉的边, 拆掉边后,原图变成2颗树,然后重新连接两颗子树的某两个结点, 那么,此时答案(新树的直径)只有可能有3种情况。
1、直径完全在1子树中。
2、直径完全在2子树中。
3、直径经过了 重新建立的那条边,也就是贯穿了2颗子树。
用这3种的最大值更新ans。
求树的直径的方法可以通过dfs或者bfs,O(n)时间求出。
现在的问题在于,如何求第3种。
假设1子树存在一个点,这个点满足,任意一个点到它的距离的最大值最小。同样,2子树也找到这样一个点,然后连接这2个点,这样求得的最长链一定是最短的。
问题转化为,如何求树上的一个点,使得任意一个点到它的距离的最大值最小
假设这样一个点a不在直径上,那么它到最远距离的点,一定会和直径产生一个交点b(由直径的性质),那么a到其他点(设为x)的最大距离一定大于b到其他点(直径的端点,设为y)的最大距离。所以a一定在直径上。
证明:其实x点就是直径的端点,因为,若x不是直径的端点,那么就有ax>ay ==>bx>by==>那么y就不应该是直径的端点了。而是x。所以x一定是直径的端点。。所以ax=ab+bx,所以ax>bx。
实际上还有一个优化,就是枚举边的时候,不必枚举每一条边,只需要枚举原图的直径上的边就可以了,因为只有删掉原图的直径边,才能使直径变短。这个优化可以把时间从4800+ms降到90+ms。
题解:枚举原图直径上的边,分别求出拆掉这条边后,1,2两颗子树的直径l1,l2,再利用上述的证明求出l3
ans = min(ans,max(l1,l2,l3));复杂度为O(n^2)
我把题目中的dfs都用一个函数写下来了,所以主函数看上去比较冗长。。
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<vector>
using namespace std;
int n;
vector<int> tree[2600];
int maps[2600][2600];
int dp[2600];
int fat[2600];
int FA[2600];
int maxn,node,ans;
void input()
{
int a,b,c;
scanf("%d",&n);
for(int i=0;i<=n;i++)
{
tree[i].clear();
}
for(int i=1;i<n;i++)
{
scanf("%d%d%d",&a,&b,&c);
tree[a].push_back(b);
tree[b].push_back(a);
maps[a][b]=maps[b][a]=c;
}
}
void dfs(int now,int NOT,int fa,int num=0)
{
if(fa==-1) node=now;
for(int i=0;i<tree[now].size();i++)
{
int to=tree[now][i];
if(to!=fa&&to!=NOT)
{
fat[to]=now;
dp[to]=dp[now]+maps[now][to];
dfs(to,NOT,now);
}
}
if(dp[now]>maxn)
{
maxn=dp[now];
node=now;
}
}
int main()
{
int cas,a,b,c;
cin>>cas;
int ca=1;
while(cas--)
{
input();
ans=0x3f3f3f3f;
dfs(0,-1,-1);
int aa=node;
dfs(aa,-1,-1);
int bb=node;
for(int i=0;i<n;i++) FA[i]=fat[i];
while(bb!=aa)
{
int i=bb;
int l1,l2;
int t1=0x3f3f3f3f,t2=0x3f3f3f3f;
maxn=0;dp[i]=0;dfs(i,FA[bb],-1);int a=node;
maxn=0;dp[a]=0;dfs(a,FA[bb],-1);int b=node;
l1=maxn;
if(a==b)
{
l1=0;
t1=0;
}
else
{
while(b!=a)
{
t1=min(t1,max(dp[b],maxn-dp[b]));
b=fat[b];
}
}
maxn=0;dp[FA[bb]]=0;dfs(FA[bb],i,-1);a=node;
maxn=0;dp[a]=0;dfs(a,i,-1);b=node;
l2=maxn;
if(a==b)
{
l2=0;
t2=0;
}
else
{
while(b!=a)
{
t2=min(t2,max(dp[b],maxn-dp[b]));
b=fat[b];
}
}
ans=min( ans, max(max(l1,l2),t1+t2+maps[i][FA[bb]]) );
bb=FA[bb];
}
printf("Case %d: %d\n",ca++,ans);
}
return 0;
}