Wannafly挑战赛1 C:MMSet2

题面

最优的点一定在给定点集中最远的两点的简单路径上。

#include<bits/stdc++.h>
using namespace std;
const int maxn=3e5+10;
int head[maxn],To[maxn*2],Next[maxn*2],tol;
void add_edge(int x,int y)
{
    To[tol]=y;
    Next[tol]=head[x];
    head[x]=tol++;
}
int lg[maxn],f[maxn][22],dep[maxn];
void dfs(int v,int fa)
{
    dep[v]=dep[fa]+1;
    f[v][0]=fa;
    for(int i=1;(1<<i)<=dep[v];i++)
        f[v][i]=f[f[v][i-1]][i-1];
    for(int i=head[v];i!=-1;i=Next[i])
    {
        int u=To[i];
        if(u==fa) continue;
        dfs(u,v);
    }
}
int lca(int x,int y)
{
    if(dep[x]>dep[y]) swap(x,y);
    while(dep[x]<dep[y])
    {
        int h=lg[dep[y]-dep[x]];
        y=f[y][h];
    }
    if(x==y) return x;
    for(int i=lg[dep[x]];i>=0;i--)
    {
        if(f[x][i]!=f[y][i])
        {
            x=f[x][i];
            y=f[y][i];
        }
    }
    return f[x][0];
}
int s[1000005],len;
int get_dist(int st,int &d)
{
    int id=st;d=0;
    for(int i=1;i<=len;i++)
    {
        int ff=lca(st,s[i]);
        int dist=dep[st]+dep[s[i]]-2*dep[ff];
        if(dist>d)
        {
            d=dist;
            id=s[i];
        }
    }
    return id;
}
int main()
{
    for(int i=2;i<maxn;i++)
        lg[i]=lg[i/2]+1;
    memset(head,-1,sizeof(head));
    int n,Q;scanf("%d",&n);
    for(int i=1,x,y;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        add_edge(x,y);
        add_edge(y,x);
    }
    dfs(1,0);
    scanf("%d",&Q);
    while(Q--)
    {
        scanf("%d",&len);
        for(int i=1;i<=len;i++)
            scanf("%d",&s[i]);
        int id,d;
        id=get_dist(s[1],d);
        get_dist(id,d);
        printf("%d\n",(d+1)/2);
    }
}