虚树第一题
对于询问多,但是询问涉及的总点数少的题型考虑建一棵虚树。
懒得讲解题目
dtoj过了,洛谷T了不管了
以下代码:
![](https://i-blog.csdnimg.cn/blog_migrate/8f900a89c6347c561fdf2122f13be562.gif)
![](https://i-blog.csdnimg.cn/blog_migrate/961ddebeb323a10fe0623af514929fc1.gif)
#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; }