牛客多校第四场 A meeting 树的半径

时间:2021-07-27 05:29:38

题意:

有一棵树,树上有许多人,他们要聚会,找一个点使得所有人到这个点的距离的最大值最小。

题解:

首先,以一个有人的点为根,求一个生成树,删掉所有没有人的子树,保证所有的悬挂点(只连接一条边的点)都是有人的节点,以保证后面求出的直径的两端是两个有人节点。为什么非得以有人的节点为根呢?因为如果找了一个没人的节点当根,而这个根又恰好是一个悬挂点,那么可以想象,这个悬挂点可能成为直径的端点。

其次,求树的直径,就是树上两点的最长距离,这个直径的两个端点必然都是悬挂点。

直径/2向上取整就是半径,也就是题目所求距离。

求直径可跑两遍dfs,第一遍从任意点开始,找出与此点距离最大点,第二遍从这个点开始,找出与这个点距离最大点。

赛时想的是树的重心,wa了,想像一棵树上半部分是一条长链,下半部分是密密麻麻的儿子接在长链一端。然后又想了拓扑求直径,T了。

但是发现拓扑排序的轮数其实就是树的半径,不知道这个结论有什么用。

#include<iostream>
#include<vector>
#define INF 0x3f3f3f3f
#include<queue>
#include<cstring>
#include<set>
using namespace std;
vector<int> lnk[];
vector<int> sclnk[];//生成树
int n; int scn;//生成树的节点数
bool ren[];
bool dfs(int fa,int u){
bool ok=ren[u];
for(int i=;i<lnk[u].size();i++){
int v=lnk[u][i];
if(v==fa)continue;
bool tr=dfs(u,v);
if(tr){
sclnk[u].push_back(v);
sclnk[v].push_back(u);
scn++;
}
ok=ok||tr;
}
return ok;
}
int kg;
void prt(int fa,int u){
//把生成树打印出来
for(int i=;i<=kg;i++)printf(" ");
printf("%d\n",u);
kg++;
for(int i=;i<sclnk[u].size();i++){
int v=sclnk[u][i];
if(v==fa)continue;
prt(u,v);
}
kg--;
} bool vis[];
int size[]; int dis[];
struct P{
int n,dis;
P(){}
P(int n1,int dis1){
n=n1;dis=dis1;
}
friend bool operator >(const P &a,const P &b){
return a.dis>b.dis;
}
friend bool operator <(const P &a,const P &b){
return a.dis<b.dis;
}
}; void ddfs(int st) {
vis[st] = ;
for(int i = ;i<sclnk[st].size();i++) {
int to =sclnk[st][i];
if(!vis[to]) {
dis[to] = dis[st] +;
ddfs(to);
}
}
} int main(){
//把一个有人的点揪出来当树根
//删掉无人的子树
//找出树的直径
int k;
scanf("%d %d",&n,&k);
for(int i=;i<n;i++){
int a,b;
scanf("%d %d",&a,&b);
lnk[a].push_back(b);
lnk[b].push_back(a);
}
int r1;
for(int i=;i<=k;i++){
int q;
scanf("%d",&q);
r1=q;
ren[q]=;
}
dfs(-,r1);
// prt(r1);
//求生成树 memset(dis,INF,sizeof dis);
dis[r1] = ;
ddfs(r1);
//求树的直径
int maxi=r1;
for(int i=;i<=n;i++){
if(dis[i]!=INF && dis[i]>dis[maxi])maxi=i;
} memset(dis,INF,sizeof dis);
memset(vis,,sizeof vis);
dis[maxi] = ;
ddfs(maxi); int maxx=;
for(int i=;i<=n;i++){
// printf("%d ",dis[i]);
if(dis[i]!=INF)maxx=max(maxx,dis[i]);
}
printf("%d\n",(maxx+)/);
return ;
}