【HNOI2014】bzoj3572 世界树

Description

世界树是一棵无比巨大的树，它伸出的枝干构成了整个世界。在这里，生存着各种各样的种族和生灵，他们共同信奉着绝对公正公平的女神艾莉森，在他们的信条里，公平是使世界树能够生生不息、持续运转的根本基石。
世界树的形态可以用一个数学模型来描述：世界树中有n个种族，种族的编号分别从1到n，分别生活在编号为1到n的聚居地上，种族的编号与其聚居地的编号相同。有的聚居地之间有双向的道路相连，道路的长度为1。保证连接的方式会形成一棵树结构，即所有的聚居地之间可以互相到达，并且不会出现环。定义两个聚居地之间的距离为连接他们的道路的长度；例如，若聚居地a和b之间有道路，b和c之间有道路，因为每条道路长度为1而且又不可能出现环，所卧a与c之间的距离为2。
出于对公平的考虑，第i年，世界树的国王需要授权m[i]个种族的聚居地为临时议事处。对于某个种族x（x为种族的编号），如果距离该种族最近的临时议事处为y（y为议事处所在聚居地的编号），则种族x将接受y议事处的管辖（如果有多个临时议事处到该聚居地的距离一样，则y为其中编号最小的临时议事处）。
现在国王想知道，在q年的时间里，每一年完成授权后，当年每个临时议事处将会管理多少个种族（议事处所在的聚居地也将接受该议事处管理）。
现在这个任务交给了以智慧著称的灵长类的你：程序猿。请帮国王完成这个任务吧。 Input

第一行为一个正整数n，表示世界树中种族的个数。 接下来n-l行，每行两个正整数x，y，表示x聚居地与y聚居地之间有一条长度为1的双
向道路。接下来一行为一个正整数q，表示国王询问的年数。 接下来q块，每块两行：
第i块的第一行为1个正整数m[i]，表示第i年授权的临时议事处的个数。
第i块的第二行为m[i]个正整数h[l]、h[2]、…、h[m[i]]，表示被授权为临时议事处的聚居地编号（保证互不相同）。 Output

输出包含q行，第i行为m[i]个整数，该行的第j(j=1，2…，，m[i])个数表示第i年被授权的聚居地h[j]的临时议事处管理的种族个数。

建好虚树，求出每个点的$up_i$和$down_i$表示向下走和向上走【这里的向上走不一定是第一步向上，也可能是走到其他子树】的最近关键点。然后对于虚树上的每一条路径$u-v$切开分给$up_u$和$down_v$。
虽然是一眼题，但是需要注意的细节非常多。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=300010,oo=0x3f3f3f3f;
int rd()
{
    int x=0;
    char c=getchar();
    while (c<'0'||c>'9') c=getchar();
    while (c>='0'&&c<='9')
    {
        x=x*10+c-'0';
        c=getchar();
    }
    return x;
}
int fir0[maxn],ne0[2*maxn],to0[2*maxn],dep[2*maxn],size[2*maxn],anc[maxn][22],
fir1[maxn],ne1[2*maxn],to1[2*maxn],
up[maxn],down[maxn],disup[maxn],disdown[maxn],left[maxn],
last[maxn],a[maxn],tem[maxn],ans[maxn],sta[maxn],mark[maxn],
log[2*maxn],pos[maxn],ord[2*maxn][22],
n,m,q,num,clo,K;
void pause()
{
    int x;
    x=1;
}
void check()
{
    for (int i=1;i<=n;i++)
        for (int j=fir1[i];j;j=ne1[j])
            printf("%d->%d\n",i,to1[j]);
}
int cmpd(int u,int v)
{
    return dep[u]<dep[v];
}
int cmpp(int u,int v)
{
    return pos[u]<pos[v];
}
void add0(int num,int u,int v)
{
    ne0[num]=fir0[u];
    fir0[u]=num;
    to0[num]=v;
}
void dfs0(int u,int fa)
{
    int v;
    ord[pos[u]=++clo][0]=u;
    size[u]=1;
    for (int i=fir0[u];i;i=ne0[i])
        if ((v=to0[i])!=fa)
        {
            anc[v][0]=u;
            dep[v]=dep[u]+1;
            dfs0(v,u);
            size[u]+=size[v];
            ord[++clo][0]=u;
        }
}
void init()
{
    int u,v;
    n=rd();
    for (int i=1;i<n;i++)
    {
        u=rd();
        v=rd();
        add0(i<<1,u,v);
        add0(i<<1|1,v,u);
    }
    dfs0(1,-1);
    for (int k=1;(1<<k)<=clo;k++) log[1<<k]=k;
    for (int i=3;i<=clo;i++)
        if (!log[i]) log[i]=log[i-1];
    for (int k=1;(1<<k)<=clo;k++)
        for (int i=1;i+(1<<k)-1<=clo;i++)
            ord[i][k]=min(ord[i][k-1],ord[i+(1<<k-1)][k-1],cmpd);
    for (int k=1;(1<<k)<=n;k++)
        for (int i=1;i<=n;i++)
            anc[i][k]=anc[anc[i][k-1]][k-1];
}
int lca(int u,int v)
{
    int x=pos[u],y=pos[v];
    if (x>y) swap(x,y);
    int k=log[y-x+1];
    return min(ord[x][k],ord[y-(1<<k)+1][k],cmpd);
}
void add1(int u,int v)
{
    num++;
    if (last[u]<K)
    {
        last[u]=K;
        fir1[u]=0;
    }
    if (last[v]<K)
    {
        last[v]=K;
        fir1[v]=0;
    }
    ne1[num<<1]=fir1[u];
    fir1[u]=num<<1;
    to1[num<<1]=v;
    ne1[num<<1|1]=fir1[v];
    fir1[v]=num<<1|1;
    to1[num<<1|1]=u;
}
void renew(int &x,int &y,int x1,int y1)
{
    if (y1<y||(y1==y&&x1<x))
    {
        x=x1;
        y=y1;
    }
}
int dis(int u,int v)
{
    int x=lca(u,v);
    return dep[u]+dep[v]-2*dep[x];
}
int fat(int u,int x)
{
    int lim=1;
    //if (x>10000000) printf("no\n");
    for (int k=0;(1<<k)<=x;k++) lim=k+1;
    //int lim=20;
    /*if (lim>100)
    {
        pause();
    }*/
    for (int k=lim;k>=0;k--)
        if ((1<<k)&x) u=anc[u][k];
    return u;
}
void dfs1(int u,int fa)
{
    int v,w;
    if (mark[u]==K) down[u]=u,disdown[u]=0;
    else down[u]=-1,disdown[u]=oo;
    left[u]=size[u];
    for (int i=fir1[u];i;i=ne1[i])
        if ((v=to1[i])!=fa)
        {
            dfs1(v,u);
            w=fat(v,dep[v]-dep[u]-1);
            left[u]-=size[w];
            if (down[v]!=-1) renew(down[u],disdown[u],down[v],dep[down[v]]-dep[u]);
        }
}
void dfs2(int u,int fa)
{
    int v,x,y,p,q,t,w;
    if (mark[u]==K) up[u]=u,disup[u]=0;
    else
    {
        x=fa!=-1&&up[fa]!=-1?dis(u,up[fa]):oo;
        y=down[u]!=-1?dis(u,down[u]):oo;
        up[u]=down[u];
        disup[u]=y;
        renew(up[u],disup[u],up[fa],x);
    }
    if (disup[u]<disdown[u]||(disup[u]==disdown[u]&&up[u]<down[u])) ans[up[u]]+=left[u];
    else ans[down[u]]+=left[u];
    for (int i=fir1[u];i;i=ne1[i])
        if ((v=to1[i])!=fa)
        {
            x=dep[v]-dep[u]-1;
            w=fat(v,x);
            if (up[u]==-1||up[u]==down[v]) ans[down[v]]+=size[w]-size[v];
            else if (down[v]==-1) ans[up[u]]+=size[w]-size[v];
            else
            {
                y=disup[u]+disdown[v]+x;
                p=q=y/2;
                if (y&1)
                {
                    if (up[u]<down[v]) p++;
                    else q++;
                }
                t=fat(v,max(0,min(q-disdown[v],x)));
                ans[down[v]]+=size[t]-size[v];
                ans[up[u]]+=size[w]-size[t];
            }
            dfs2(v,u);
        }
}
void solve()
{
    int top,u;
    m=rd();
    for (int i=1;i<=m;i++)
        mark[tem[i]=a[i]=rd()]=K;
    sort(a+1,a+m+1,cmpp);
    sta[top=1]=1;
    num=0;
    for (int i=a[1]==1?2:1;i<=m;i++)
    {
        u=lca(a[i],sta[top]);
        while (dep[sta[top]]>dep[u])
        {
            add1(sta[top],max(sta[top-1],u,cmpd));
            top--;
        }
        if (u!=sta[top]) sta[++top]=u;
        sta[++top]=a[i];
    }
    for (;top>1;top--) add1(sta[top],sta[top-1]);
    //check();
    for (int i=1;i<=m;i++) ans[a[i]]=0;
    dfs1(1,-1);
    dfs2(1,-1);
    for (int i=1;i<=m;i++) printf("%d ",ans[tem[i]]);
    putchar('\n');
}
int main()
{
    /*freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);*/
    init();
    q=rd();
    for (K=1;K<=q;K++) solve();
}