2018.12.13-dtoi1877-世界树(worldtree)

虚树第一题

对于询问多，但是询问涉及的总点数少的题型考虑建一棵虚树。

懒得讲解题目

dtoj过了，洛谷T了不管了

以下代码：

#include<bits/stdc++.h>
#define il inline
#define _(d) while(d(isdigit(ch=getchar())))
using namespace std;
const int N=3e5+5;
int n,head[N],ne[N<<1],to[N<<1],cnt,id[N],d[N],tot,fa[N][20];
int sz[N],m,a[N],b[N],q[N],c[N],rem[N],dfn,bel[N],f[N];
il int read(){int x;char ch;_(!);x=ch^48;_()x=(x<<1)+(x<<3)+(ch^48);return x;}
il void insert(int x,int y){ne[++cnt]=head[x];head[x]=cnt;to[cnt]=y;}
bool cmp(int t1,int t2){return id[t1]<id[t2];}
il void dfs(int x){
    id[x]=++tot;sz[x]=1;
    for(int i=1;fa[x][i-1];i++)fa[x][i]=fa[fa[x][i-1]][i-1];
    for(int i=head[x];i;i=ne[i]){
        if(fa[x][0]==to[i])continue;
        fa[to[i]][0]=x;d[to[i]]=d[x]+1;
        dfs(to[i]);sz[x]+=sz[to[i]];
    }
}
il int Lca(int x,int y){
    if(d[x]<d[y])swap(x,y);
    for(int i=18;i>=0;i--)if(d[fa[x][i]]>=d[y])x=fa[x][i];
    if(x==y)return x;
    for(int i=18;i>=0;i--)if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
il int dis(int x,int y){return d[x]+d[y]-2*d[Lca(x,y)];}
il void dfs1(int x){
    c[++dfn]=x;rem[x]=sz[x];
    for(int i=head[x];i;i=ne[i]){
        dfs1(to[i]);
        if(!bel[to[i]])continue;
        int t1=dis(bel[to[i]],x),t2=dis(bel[x],x);
        if(t1<t2||(t1==t2&&bel[to[i]]<bel[x])||!bel[x])bel[x]=bel[to[i]];
    }
}
il void dfs2(int x){
    for(int i=head[x];i;i=ne[i]){
        int t1=dis(bel[x],to[i]),t2=dis(bel[to[i]],to[i]);
        if(t1<t2||(t1==t2&&bel[x]<bel[to[i]])||!bel[to[i]])bel[to[i]]=bel[x];
        dfs2(to[i]);
    }
}
il void solve(int a,int b){
    int x=b,mid=b;
    for(int i=18;i>=0;i--)
        if(d[fa[x][i]]>d[a])x=fa[x][i];
    rem[a]-=sz[x];
    if(bel[a]==bel[b]){
        f[bel[a]]+=sz[x]-sz[b];return;
    }
    for(int i=18;i>=0;i--){
        int nxt=fa[mid][i];
        int t1=dis(bel[a],nxt),t2=dis(bel[b],nxt);
        if(t1>t2||(t1==t2&&bel[b]<bel[a]))mid=nxt;
    }
    f[bel[a]]+=sz[x]-sz[mid];
    f[bel[b]]+=sz[mid]-sz[b];
}
il void query(){
    m=read();tot=0;dfn=0;cnt=0;
    for(int i=1;i<=m;i++)a[i]=b[i]=read();
    for(int i=1;i<=m;i++)bel[a[i]]=a[i];
    sort(a+1,a+1+m,cmp);if(bel[1]!=1)q[++tot]=1;
    for(int i=1;i<=m;i++){
        int t=a[i],lca=-1;
        while(tot>0){
            lca=Lca(t,q[tot]);
            if(tot>1&&d[q[tot-1]]>d[lca])insert(q[tot-1],q[tot]),tot--;
            else if(d[q[tot]]>d[lca]){insert(lca,q[tot--]);break;}
            else break;
        }
        if(lca!=q[tot])q[++tot]=lca;q[++tot]=t;
    }
    while(tot>1)insert(q[tot-1],q[tot]),tot--;
    dfs1(1);dfs2(1);
    for(int i=1;i<=dfn;i++)
        for(int j=head[c[i]];j;j=ne[j])solve(c[i],to[j]);
    for(int i=1;i<=dfn;i++)f[bel[c[i]]]+=rem[c[i]];
    for(int i=1;i<=m;i++)printf("%d ",f[b[i]]);puts("");
    for(int i=1;i<=dfn;i++)
        f[c[i]]=bel[c[i]]=head[c[i]]=0;
}
int main()
{
    n=read();
    for(int i=1;i<n;i++){int x=read(),y=read();insert(x,y);insert(y,x);}
    d[1]=1;dfs(1);int Q=read();for(int i=1;i<=n;i++)head[i]=0;
    while(Q--)query();
  return 0;
}