Building Fire Stations

题目链接

首先找到树的直径，直径左端点是a，直径右端点是b，中间的点是mid（偶数的情况下mid可以看做两个），两点因该是左右分布；

假设两点都不在直径上，那么移到直径上的话距离更短；

设直径上左边的消防站是x，右边的消防站是y；

如果直径上的两个点不对称的话，那个距离就肯定大于等于a到x，b到y；所以可以让过度的点(已经满足要求)一点，那样更优而且并不影响；

直径上的点更靠近中间的话，就更符合要求但是距离也会变大，就是单调的图，由不行变成行；

现在就是要求行的分界线；

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<queue>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const int N=2e5+10,mod=1e6+3;
int h[N],ne[N<<1],e[N<<1],idx=0;//邻接表寸土 
void add(int a,int b){ne[idx]=h[a],e[idx]=b,h[a]=idx++;return ;}
int dis[N],dis1[N],dis2[N],n;//dist-直径 dist1-左边点 dist2-右边点； 
vector<int>v;//存直径 
void dfs(int u,int fa,int dis[])//距离 
{
	dis[u]=dis[fa]+1;
	for(int i=h[u];~i;i=ne[i])
	{
		int t=e[i];
		if(t==fa)continue;
		dfs(t,u,dis);
	}
	return ;
}
void zhijing()//找直径上的点 
{
	memset(dis,0,(n+5)*sizeof(int));
	dfs(1,0,dis);
	int ma=0;
	for(int i=2;i<=n;i++)if(dis[ma]<dis[i])ma=i;
	dfs(ma,0,dis);
	for(int i=1;i<=n;i++)if(dis[ma]<dis[i])ma=i;
	int sum=dis[ma];
	while(sum)
	{
		for(int i=h[ma];~i;i=ne[i])
			if(dis[e[i]]==sum)
			{ma=e[i];break;}
		v.push_back(ma);
		sum--;
	}
	return ;
}

int main()
{
	int t;
	scanf("%d",&t);
	for(int _=1;_<=t;_++)
	{
		scanf("%d",&n);
		idx=0;
		memset(h,-1,(n+5)*sizeof(int)); 
		for(int i=1,l,r;i<n;i++)scanf("%d%d",&l,&r),add(l,r),add(r,l);
		v.clear();
		zhijing();
		int l=0,siz=v.size()-1,r=siz/2,ans=0;
		while(l<r)
		{
			int mid=l+r>>1,ma=0;
			int a=v[mid],b=v[siz-mid];
			dfs(a,0,dis1);
			dfs(b,0,dis2);
			for(int i=1;i<=n;i++)ma=max(ma,min(dis1[i],dis2[i]));
			if(ma<=mid+1)r=mid;
			else l=mid+1;
		}
		dfs(v[l],0,dis1);
		dfs(v[siz-l],0,dis2);
		for(int i=1;i<=n;i++)ans=max(ans,min(dis1[i],dis2[i]));
		ans--;
		if(v.size()%2&&l==siz/2)r=siz;
		else r=siz-l;
		l=v[l],r=v[r];
		printf("%d %d %d",ans,l,r);
		if(_!=t)puts("");
	} 
	return 0;
}
/*
1
13
1 2
2 3
3 4
3 5
5 7
5 6
5 9
9 8
9 10
10 11
10 12
7 13
一个样例 
*/