luogu P3761 [TJOI2017]城市

背景：

上周为了那个勤写标兵$\text{Lv3}$没有写博客，现在补补坑。
现在也没有拿到，差评。

题目传送门：

https://www.luogu.org/problemnew/show/P3761

题意：

一棵树，现在你可以断掉一条边，然后将其修在其它位置（必须保证修完后任意两点仍联通，还是一棵树），现在要使得修完的树的直径尽可能小。

思路：

大力观察数据范围，发现$n\le 5000$，因此$\Theta (n^2)$的可以过。
不妨枚举断掉哪一条边。
问题就转化成了如何连接两棵树使得形成的新的树的直径尽可能小。
考虑新的树的直径的来源。
$[1]$：原来两棵树的直径更新得到；
$[2]$：原来两棵树的半径+当前这一条边的长度（最远点对到树的中心的距离的最大值，树的中心是树的直径的距离上的中点）。
求树的直径的一种显然的做法就是从任意一点向外跑，找到距离最远的一点，再向外跑，找到距离最远的一点，这两个点就是树的最远点对，两点间的距离就是树的直径的大小。
那么树的半径怎么得到？可以记录一下找树的直径时的父亲，暴力跳父亲，知道第一次大于$\frac{1}{2}$树的直径即可，所经过的长度就是树的半径的大小。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
	int n,len,ans=2147483647;
struct node1{int x,y,z,next;} a[10010];
	struct node2{int x,y,z;} b[10010];
	int last[10010],dis[10010];
void ins(int x,int y,int z)
{
	a[++len]=(node1){x,y,z,last[x]}; last[x]=len;
}
int Fa[10010];
void dfs(int x,int fa)
{
	for(int i=last[x];i;i=a[i].next)
	{
		int y=a[i].y;
		if(y==fa) continue;
		dis[y]=dis[x]+a[i].z;
		Fa[y]=x;
		dfs(y,x);
	}
}
int get_r(int st,int ed,int d)
{
	if(st==ed) return 0;
	if(Fa[ed]==st) return d;
	for(;st!=ed;ed=Fa[ed])
		if(dis[ed]<=d/2) return d-dis[ed];
	return 0;
}
int d[3],r[3];
void get(int id,int x)
{
	int ma,st=x,ed=x;
	memset(dis,0,sizeof(dis));
	ma=dis[st]=0;
	dfs(st,0);
	for(int i=1;i<=n;i++)
		if(dis[i]>ma) ma=dis[i],st=i;
	memset(dis,0,sizeof(dis));
	ma=dis[st]=0;
	dfs(st,0);
	for(int i=1;i<=n;i++)
		if(dis[i]>ma) ma=dis[i],ed=i;
	d[id]=ma;
	r[id]=get_r(st,ed,d[id]);
	//printf("A,B=%d %d\n",st,ed);
}
int solve(int x,int y,int z)
{
	get(1,x),get(2,y);
	//printf("%d %d %d %d\n\n",d[1],d[2],r[1],r[2]);
	return max(max(d[1],d[2]),r[1]+r[2]+z);
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<n;i++)
		scanf("%d %d %d",&b[i].x,&b[i].y,&b[i].z);
	for(int i=1;i<n;i++)
	{
		len=0;
		memset(last,0,sizeof(last));
		for(int j=1;j<n;j++)
			if(i!=j) ins(b[j].x,b[j].y,b[j].z),ins(b[j].y,b[j].x,b[j].z);
		ans=min(ans,solve(b[i].x,b[i].y,b[i].z));
	}
	printf("%d",ans);
}