2019牛客暑期多校训练营（第四场）A题 树的直径

题目链接：https://ac.nowcoder.com/acm/contest/884/A

两边bfs  第一遍求任意所选节点到另一所选节点的距离最大的那个点(假设是a)，第二遍求a到另一所选节点的距离最大值，也就是树的直径

AC代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<queue>
#include<cmath>
#include<string>
#include<set>
#include<map>
using namespace std;
#define LL long long
const int MOD=1e9+7;
const int inf=0x3f3f3f3f;
const LL inff=0x3f3f3f3f3f3f3f3f;
const LL MAX_N=10005;
const LL MAX_M=100005;
#define MEF(x) memset(x,-1,sizeof(x))
#define ME0(x) memset(x,0,sizeof(x))
#define MEI(x) memset(x,inf,sizeof(x))
struct Edge
{
    int v,next;
}edge[200005];
int cnt,first[100005],vis1[100005];
void init()
{
    cnt=0;
    MEF(first);
}
void add(int u,int v)
{
    edge[cnt].v=v;
    edge[cnt].next=first[u];
    first[u]=cnt++;
}
int bfs(int x,int y)
{
    int vis[100005],ans1=x,ans2=0;
    ME0(vis);
    queue<pair<int,int> > q;
    q.push(make_pair(x,0));
    vis[x]=1;
    while(!q.empty())
    {
        int lx=q.front().first,ly=q.front().second;
        if(vis1[lx])
        {
            ans1=lx;
        }
        if(vis1[lx])
        {
            ans2=ly;
        }
        q.pop();
        for(int i=first[lx];i!=-1;i=edge[i].next)
        {
            int lv=edge[i].v;
            if(!vis[lv])
            {
                q.push(make_pair(lv,ly+1));
                vis[lv]=1;
            }
        }
    }
    if(!y)
    {
        return ans1;
    }
    else
    {
        return ans2;
    }
}
int main()
{
    ios::sync_with_stdio(false);
    int n,k,al[100005];
    cin>>n>>k;
    init();
    for(int i=1,a,b;i<=n-1;i++)
    {
        cin>>a>>b;
        add(a,b);
        add(b,a);
    }
    ME0(vis1);
    for(int i=1;i<=k;i++)
    {
        cin>>al[i];
        vis1[al[i]]=1;
    }
    int re=bfs(al[1],0);
    re=bfs(re,1);
    cout<<re/2+re%2<<endl;
    return 0;
}