[ BZOJ3572 ] [ Hnoi2014 ] [ 虚树 ] [ 树形DP ] 世界树

学一发虚树
这个树形DP是真的烦，都调发烧了……我说真的

#include <cstdio>
#include <iostream>
#include <algorithm>

using namespace std;

const int N=300010;

int n,cnt,x,y,tms;
int G[N],dfn[N],size[N];

struct edge{
  int t,nx;
}E[N<<1];

inline char nc(){
  static char buf[100000],*p1=buf,*p2=buf;
  return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}

inline void rea(int &x){
  char c=nc(); x=0;
  for(;c>'9'||c<'0';c=nc());for(;c>='0'&&c<='9';x=x*10+c-'0',c=nc());
}

inline void add(int x,int y){
  E[++cnt].t=y; E[cnt].nx=G[x]; G[x]=cnt;
  E[++cnt].t=x; E[cnt].nx=G[y]; G[y]=cnt;
}

int fa[N][30],depth[N],lab[N],near[N];

void dfs(int x,int p){
  fa[x][0]=p; dfn[x]=++tms;
  depth[x]=depth[p]+1; size[x]=1;
  for(int i=1;i<=18;i++) fa[x][i]=fa[fa[x][i-1]][i-1];
  for(int i=G[x];i;i=E[i].nx)
    if(E[i].t!=p) dfs(E[i].t,x),size[x]+=size[E[i].t];
}

inline bool cmp(const int &a,const int &b){
  return dfn[a]<dfn[b];
}

inline int lca(int x,int y){
  if(depth[x]<depth[y]) swap(x,y);
  for(int i=18;~i;i--)
    if(depth[fa[x][i]]>=depth[y]) x=fa[x][i];
  if(x==y) return x;
  for(int i=18;~i;i--)
    if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
  return fa[x][0];
}

inline int nfather(int x,int dpt){
  dpt=depth[x]-dpt;
  for(int i=18;~i;i--)
    if(depth[fa[x][i]]>=dpt) x=fa[x][i];
  return x;
}

inline int dist(int x,int y){
    return depth[x]+depth[y]-2*depth[lca(x,y)];
}

struct iTree{
  int G[N],S[N],cnt,top,root,vis[N];
  edge E[N<<1];
  void add(int x,int y){
    E[++cnt].t=y; E[cnt].nx=G[x]; G[x]=cnt;
    E[++cnt].t=x; E[cnt].nx=G[y]; G[y]=cnt;
  }
  void build(int *a,int n){
    cnt=top=0;
    for(int i=1;i<=n;i++) G[a[i]]=0,vis[a[i]]=tms;
    G[1]=0; vis[1]=tms; near[1]=0;
    S[top=1]=1;
    if(a[1]^1) S[++top]=a[1];
    for(int i=2;i<=n;i++){
      int LCA=lca(a[i],S[top]);
      if(vis[LCA]!=tms) vis[LCA]=tms,G[LCA]=near[LCA]=0;
      while(top&&depth[LCA]<depth[S[top]]){
        if(top>1&&depth[LCA]>=depth[S[top-1]]){
          add(LCA,S[top]);
          if(S[--top]!=LCA) S[++top]=LCA;
          break;
        }
        top--; add(S[top],S[top+1]);
      }
      S[++top]=a[i];
    }
    while(--top) add(S[top],S[top+1]);
    }
}iT;

int q,m;
int a[N],b[N],ans[N];

#define iG iT.G[x]
#define nxt iT.E[i].nx
#define V iT.E[i].t

inline bool closer(int x,int y,int a){
    if(!x) return false;
    if(!y) return true;
    return dist(x,a)<dist(y,a)|| (dist(x,a)==dist(y,a)&&x<y);
}

void dp1(int x,int p){
    for(int i=iG;i;i=nxt)
        if(V!=p){
            dp1(V,x);
            if(closer(near[V],near[x],x)) near[x]=near[V];  
        }
    if(lab[x]==tms) near[x]=x;
}

void dp2(int x,int p){
    if(p&&closer(near[p],near[x],x)) near[x]=near[p];
    for(int i=iG;i;i=nxt)
        if(V!=p) dp2(V,x);
}

void dp3(int x,int p){
    int sz=size[x];
    for(int i=iG;i;i=nxt)
        if(V!=p) sz-=size[nfather(V,depth[V]-depth[x]-1)];
    ans[near[x]]+=sz;
    for(int i=iG;i;i=nxt)
        if(V!=p){
            int sz=size[nfather(V,depth[V]-depth[x]-1)];
            if(near[V]==near[x]) ans[near[x]]+=sz-size[V];
            else{
                int dis1=dist(near[V],V),dis2=dist(near[x],x),delt=dis1+dis2+depth[V]-depth[x];
                if(delt&1){
                    int cent=nfather(V,(delt>>1)-dis1);
                    ans[near[V]]+=size[cent]-size[V];
                    ans[near[x]]+=sz-size[cent];
                }
                else{
                    int cent;
                    if(near[x]<near[V]) cent=nfather(V,delt/2-1-dis1);
                    else cent=nfather(V,delt/2-dis1);
                    ans[near[V]]+=size[cent]-size[V];
                    ans[near[x]]+=sz-size[cent];
                }
            }
            sz-=size[V];
        }
    for(int i=iG;i;i=nxt)
        if(V!=p) dp3(V,x);
}

void solve(){
  ++tms;
  for(int i=1;i<=m;i++)
    b[i]=a[i],ans[a[i]]=0,lab[a[i]]=tms,near[a[i]]=0;
  sort(b+1,b+1+m,cmp);
  iT.build(b,m);
  dp1(1,0); dp2(1,0); 
    dp3(1,0);
  for(int i=1;i<=m;i++)
    printf("%d ",ans[a[i]]);
  putchar('\n');
}

int main(){
  rea(n);
  for(int i=1;i<n;i++)
    rea(x),rea(y),add(x,y);
  dfs(1,0);
  rea(q); tms=0;
  while(q--){
    rea(m);
    for(int i=1;i<=m;i++) rea(a[i]);
    solve();
  }
  return 0;
}