[ 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].tinline 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;}