题目链接:
https://ac.nowcoder.com/acm/contest/884/A
题目大意
在一个有n点的树上,有k个人,要从树上选一个点,这k个人都要到这个点来开会。树的边权都为1.
问:最少需要多长时间大家能都到一个点来开会(大家是同时出发的)
解题思路:
这个很像树的直径,随便选一个人,找离这个人最远的一个人x,然后再从x这个人开始找,找一个最远的人y,则x—>y就是最长的链,xy之间的距离的一半向上取整就是答案。
AC代码:
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5+10;
int head[maxn],cnt = 0;
struct node{
int to;
int next;
}side[maxn*2];
int x[maxn];
int n,k;
int dis[maxn];
void init(){
memset(head,-1,sizeof(head));
memset(dis,0,sizeof(dis));
cnt = 0;
}
void add(int x,int y){
side[cnt].to = y;
side[cnt].next = head[x];
head[x] = cnt++;
}
int max_id = x[0];
map<int,int> mp;
void dfs(int x,int f){
for(int i=head[x];i!=-1;i=side[i].next){
int to =side[i].to;
if(to == f) continue;
dis[to] = dis[x]+1;
if(dis[to]>dis[max_id]&&mp[to]){
max_id = to;
}
dfs(to,x);
}
}
void show(){
for(int i=1;i<=n;i++){
cout<<dis[i]<<" ";
}
cout<<endl;
}
int main(){
scanf("%d%d",&n,&k);
init();
int u,v;
for(int i=0;i<n-1;i++){
scanf("%d%d",&u,&v);
add(u,v);
add(v,u);
}
for(int i=0;i<k;i++){
scanf("%d",&x[i]);
mp[x[i]]=1;
}
dfs(x[0],-1);
memset(dis,0,sizeof(dis));
dfs(max_id,-1);
//show();
int ans = dis[max_id];
if(ans%2==0) printf("%d\n",ans/2);
else printf("%d\n",ans/2+1);
return 0;
}
/*
8 4
1 2
2 3
2 4
1 5
5 6
5 7
7 8
1 3 4 8
*/