题目链接
首先找到树的直径,直径左端点是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
一个样例
*/