题目链接:https://ac.nowcoder.com/acm/contest/884/A
用给的那些点建树,求树的直径就行
昨天就做了一个,不想搞acm了,我真佩服我竟然能坚持到现在。。。
#include <bits/stdc++.h>
using namespace std;
const int maxN=1e5+5;
int N,K,x[maxN],dis[maxN];
int head[maxN],tol;
struct Graph
{
int to,next;
} path[maxN<<1];
void add(int u,int v)
{
path[tol]= {v,head[u]};
head[u]=tol++;
}
void init()
{
memset(head,-1,sizeof(head));
tol=0;
}
void dfs(int u,int fa)
{
dis[u]=dis[fa]+1;
//printf("%d %d\n",u,fa);
for(int i=head[u];i!=-1;i=path[i].next)
{
int v=path[i].to;
if(v==fa)continue;
dfs(v,u);
}
}
int main()
{
int u,v,cur;
scanf("%d%d",&N,&K);
init();
for(int i=1; i<N; i++)
{
scanf("%d%d",&u,&v);
add(u,v);
add(v,u);
}
for(int i=1; i<=K; i++)scanf("%d",&x[i]);
cur=x[1];
dfs(cur,0);
for(int i=2; i<=K; i++)
{
if(dis[cur]<dis[x[i]])
cur=x[i];
}
dfs(cur,0);
for(int i=1; i<=K; i++)
{
if(dis[cur]<dis[x[i]])
cur=x[i];
}
dis[cur]--;
printf("%d",dis[cur]/2+dis[cur]%2);
return 0;
}