Wannafly挑战赛1 C MMSet2

题目链接：Wannafly挑战赛1 C MMSet2

题意：

中文不解释。

题解：

比赛的时候是用的lca+贪心。

今天学了学虚树，这题实际就是求一个虚树的直径。

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=(a);i<=(b);++i)
#define mst(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef pair<int,int>P;

const int N=3e5+7;
int in[N],out[N],ti,Dep[N];
namespace LCA
{
    int g[N],v[N*2],nxt[N*2],ed,f[20][N*2];
    int idx[N],dep[N*2],vs[N*2],dfn;
    void init(int n){ed=0,dfn=1;F(i,0,n)g[i]=0;}
    void adg(int x,int y){v[++ed]=y,nxt[ed]=g[x],g[x]=ed;}
    void dfs(int u=1,int fa=0,int d=0)
    {
        in[u]=++ti,Dep[u]=d;
        idx[u]=dfn,vs[dfn]=u,dep[dfn++]=d;
        for(int i=g[u];i;i=nxt[i])if(v[i]!=fa)
        dfs(v[i],u,d+1),vs[dfn]=u,dep[dfn++]=d;
        out[u]=ti;
    }
    void lca_init(int rt)
    {
        dfs(rt);F(i,1,dfn-1)f[0][i]=i;
        for(int i=1;(1<<i)<dfn;++i)
            for(int j=1;j+(1<<i)-1<dfn;++j)
            {
                int a=f[i-1][j],b=f[i-1][j+(1<<(i-1))];
                f[i][j]=dep[a]<=dep[b]?a:b;
            }
    }
    int lca(int x,int y)
    {
        x=idx[x],y=idx[y];
        if(x>y)x^=y,y^=x,x^=y;
        int k=31-__builtin_clz(y-x+1);
        int a=f[k][x],b=f[k][y-(1<<k)+1];
        return vs[dep[a]<=dep[b]?a:b];
    }
}


namespace vtree
{
    int q[N],t,tot,x;bool vis[N],vip[N];vector<P>G[N];
    void clear(int *a,int tot){F(i,1,tot)vis[a[i]]=vip[a[i]]=0,G[a[i]].clear();}
    bool cmp(int x,int y){return in[x]<in[y];}
    void addedge(int x,int y)
    {
        G[x].push_back(P(y,Dep[y]-Dep[x]));
        G[y].push_back(P(x,Dep[y]-Dep[x]));
    }
    //构建m个点的虚树，用完记得clear(a,tot),注意特判1号节点。
    int build(int *a,int m)
    {
        F(i,1,m)vis[a[i]]=vip[a[i]]=1;
        if(!vis[1])a[++m]=1;
        sort(a+1,a+1+m,cmp),tot=m;
        F(i,2,m)if((x=LCA::lca(a[i-1],a[i]))!=1&&!vis[x])vis[a[++tot]=x]=1;
        sort(a+1,a+1+tot,cmp),q[t=1]=1;
        F(i,2,tot)
        {
            while(in[a[i]]<in[q[t]]||out[a[i]]>out[q[t]])t--;
            addedge(q[t],a[i]),q[++t]=a[i];
        }
        return tot;
    }
}

int n,m,x,y,num,a[N],mx[N];

void dfs(int x,int fa,int val)
{
    mx[x]=val;
    for(auto &it:vtree::G[x])
        if(it.first!=fa)
            dfs(it.first,x,val+it.second);
}

int main(){
    scanf("%d",&n);
    F(i,2,n)
    {
        scanf("%d%d",&x,&y);
        LCA::adg(x,y),LCA::adg(y,x);
    }
    LCA::lca_init(1),scanf("%d",&m);
    F(i,1,m)
    {
        scanf("%d",&num);
        F(j,1,num)scanf("%d",a+j);
        num=vtree::build(a,num);
        dfs(1,0,0);
        int now,v=-1;
        F(j,1,num)if(mx[a[j]]>v)now=a[j],v=mx[a[j]];
        dfs(now,0,0),v=-1;
        F(j,1,num)if(vtree::vis[a[j]])
        {
            if(mx[a[j]]>v)v=mx[a[j]];
        }
        printf("%d
",(v+1)/2);
        vtree::clear(a,num);
    }
    return 0;
}