P3761 [TJOI2017]城市

题目描述

从加里敦大学城市规划专业毕业的小明来到了一个地区城市规划局工作。这个地区一共有ri座城市，《-1条高速公路，保证了任意两运城市之间都可以通过高速公路相互可达，但是通过一条高速公路需要收取一定的交通费用。小明对这个地区深入研究后，觉得这个地区的交通费用太贵。小明想彻底改造这个地区，但是由于上司给他的资源有限，因而小明现在只能对一条高速公路进行改造，改造的方式就是去掉一条高速公路，并且重新修建一条一样的高速公路(即交通费用一样），使得这个地区的两个城市之间的最大交通费用最小（即使得交通费用最大的两座城市之间的交通费用最小），并且保证修建完之后任意两座城市相互可达。如果你是小明，你怎么解决这个问题？

输入输出格式

输入格式：

输入数据的第一行为一个整数n,代表城市个数。

接下来的n - 1行分别代表了最初的n-1条公路情况。每一行都有三个整数u,v,d。u,v代表这条公路的两端城市标号，d代表这条公路的交通费用。

1 <= u,v <= n,1<= d <= 2000

输出格式：

输出数据仅有一行，一个整数，表示进行了最优的改造之后，该地区两城市 之间最大交通费用。

输入输出样例

输入样例#1： 

5
1 2 1
2 3 2
3 4 3
4 5 4

输出样例#1：

7

说明

对于30%的数据，1<=n<500

对于100%的数据，1<=n<=5000

题意理解：

大概就是给你一棵无根树，可以改变一条边的连接方式，求直径的最小值。

思路：

这条改变的边一定是在原树的直径上。因为如果不是的话，那么原树的直径是不会变小的，因此没有意义。

改变一条边产生的操作显然是先把原树分成了两棵独立的树。这时候再用那条边把两棵树以直径最小的方式连接起来。

连接两棵树的中心（到所有点最短路径的最大值最小）即可。产生的新的树的直径要么是其中一棵树的直径长度，要么是两棵树的半径长度再加上连起来的那条边。

求树的直径用两遍bfs/dfs都可以。第一遍随便从一个点开始，找出离它距离最远的点，第二遍从这个点出发，找出离它最远的点。则这个最远距离就是直径，这两个点就是直径的端点。（可以自己证明一下）

其实树的中心位于它直径的中点上。证明方法和上面那个很像。就是如果它不在直径上，那么到直径其中一个端点的距离必然大于直径上某个点到那个端点的距离。（有点像 两点之间直线最短）如果在直径上的话，显然是在直径的中点上可以保证到两个端点的距离最小。（感觉好像有点问题 谁能指出一下呀qwq）

如果直径长度是偶数的话，要分别看一看这两个点的情况。

备注：

这道题要定义好多好多个变量名，还有好多好多初始化，写的超级晕！一定要保证脑子清楚。还有这棵树可能是一条链，这样的话就没有必要再去bfs算两棵树的直径什么的了，直接枚举边就可以。这个优化可以让你的程序至少快一倍。

感觉思维上也不是很难，但是对于我来说代码比较恶心。花了好几个小时才调过（可能是因为我太菜了）

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
typedef unsigned int ui;
struct edge
{
  int to,d;
};
const int N=5010;
vector<edge> g[N];
int curmx1,curv1,curmx2,curv2;
int dis[N],dv[N],diss[N],dvv[N];
int len=0;
int n;
int hlfd;
bool b[N];
void dfs1(int x,int dep,int fa)
{
  if(dep>curmx1) curmx1=dep,curv1=x;
  for(ui i=0;i<g[x].size();++i)
  {
    int xx=g[x][i].to,dd=g[x][i].d;
    if(!b[xx] && xx != fa) dfs1(xx,dep+dd,x);
  }
}
void dfs2(int x,int dep,int fa)
{
  if(dep>curmx2) curmx2=dep,curv2=x;
  for(ui i=0;i<g[x].size();++i)
  {
    int xx=g[x][i].to,dd=g[x][i].d;
    if(!b[xx] && xx != fa) dfs2(xx,dep+dd,x);
  }
}
bool dfs3(int x,int dep,int fa)
{
   dis[++len]=dep;
   dv[len]=x;
   if(x==curv2) return 1;
   for(ui i=0;i<g[x].size();i++)
   {
     int xx=g[x][i].to,dd=g[x][i].d;
     if(!b[xx] && xx != fa) 
     {
       if(dfs3(xx,dep+dd,x)==1) return 1;
     }
   }
   --len;
   return 0;
}
inline int got(int u)
{
  dfs1(curv1=u,curmx1=0,0);
  dfs2(curv1,curmx2=0,len=0);
  dfs3(curv1,0,0);
  for(int j=1;j<=len;++j)
  {
    if(dis[j]<=(curmx2>>1) && dis[j+1]>=(curmx2>>1))
    {
      hlfd+=min(max(dis[j],curmx2-dis[j]),max(dis[j+1],curmx2-dis[j+1]));
      break;
    }
  }
  return curmx2;
}
int main()
{
  scanf("%d",&n);
  int u,v,d;
  for(int i=1;i<=n;++i)
  {
    scanf("%d%d%d",&u,&v,&d);
    g[u].push_back((edge){v,d});
    g[v].push_back((edge){u,d});
  }
  dfs1(u,curmx1=0,0);
  dfs2(curv1,curmx2=0,len=0);
  dfs3(curv1,0,0);
  int ans=1<<30;
  int lenn=len;
  for(int i=1;i<=len;++i)
  {
    diss[i]=dis[i],dvv[i]=dv[i];
  }
  int str,ed;
  if(len==n)
  {
    for(int i=2;i<=lenn;++i)
    {
      str=dvv[i-1];
      ed=dvv[i];
      curmx2=dis[str];
      hlfd=0;
      for(int j=1;j<=str;++j)
      {
        if(dis[j]<=(curmx2>>1) && dis[j+1]>=(curmx2>>1))
        {
          hlfd+=min(max(dis[j],curmx2-dis[j]),max(dis[j+1],curmx2-dis[j+1]));
          break;
        }
      }
      curmx2=dis[n]-dis[ed];
      for(int j=ed;j<=n;++j)
      {
        if(dis[j]-dis[str]<=curmx2>>1 && dis[j+1]-dis[str]>=curmx2>>1)
        {
          hlfd+=min(max(dis[j]-dis[str],curmx2-dis[j]+dis[str]),max(dis[j+1]-dis[str],curmx2-dis[j+1]+dis[str]));
          break;
        }
      }
      ans=min(ans,max(max(dis[str],dis[n]-dis[ed]),hlfd+dis[i]-dis[i-1]));
    }
    //cout<<ans<<endl;
    return printf("%d\n",ans),0;
  }
  for(int i=2;i<=lenn;++i)
  {
    str=dvv[i-1];
    ed=dvv[i];
    hlfd=0;
    b[str]=1;
    int hlf1=got(ed);
    b[str]=0;
    b[ed]=1;
    int hlf2=got(str);
    ans=min(ans,max(max(hlf1,hlf2),hlfd+diss[i]-diss[i-1]));
    b[i]=0;
  }
  return printf("%d\n",ans),0;
}