先建出虛樹(似乎虛樹可以不寫重新標號)
dfs兩次維護
考慮虛樹中的每一條邊,這條邊上的所有不在虛樹上的點以及他的子樹只能選這條邊兩端的虛樹上的點的
求出分界位置。對於求一條鏈上點的子樹的大小之和,可以倍增維護一個點到祖先的size之和,假設求鏈fa[x]到fa[y]上點的子樹之和(fa[y]是fa[x]的祖先),就是(fa[x]到fa[y]的size和)-(x到y的size和)。
再對於每個虛樹上的點單獨計算就好了。
(語文太渣,可以看代碼。。。)
#include <bits/stdc++.h>
using namespace std;
#define inf 1e9
#define N 310000
#define MP make_pair
#define PA pair<int,int>
#define AD(x,y) MP(x.first+y,x.second)
int n,tot,cnt,Q,top,tim;
int head[N],nex[N<<1],to[N<<1],pos[N],a[N],b[N];
int st[N],deep[N],bel[N],num[N],bj[N],ans[N];
int fa[N][21],sum[N][21];
vector<int>vec[N];
PA f[N],g[N];
void add(int x,int y)
{
tot++;
nex[tot]=head[x];head[x]=tot;
to[tot]=y;
}
void dfs(int x,int y)
{
pos[x]=++cnt;deep[x]=deep[y]+1;
sum[x][0]=1;fa[x][0]=y;
for(int i=head[x];i;i=nex[i])
if(to[i]!=y)
{
dfs(to[i],x);
sum[x][0]+=sum[to[i]][0];
}
}
int cmp(int x,int y){return pos[x]<pos[y];}
int lca(int x,int y)
{
if(deep[x]<deep[y])swap(x,y);
for(int i=20;i>=0;i--)
if(deep[fa[x][i]]>=deep[y])
x=fa[x][i];
if(x==y)return x;
for(int i=20;i>=0;i--)
if(fa[x][i]!=fa[y][i])
x=fa[x][i],y=fa[y][i];
return fa[x][0];
}
void dfs1(int x)
{
f[x]=MP(inf,0);
if(bj[bel[x]]==tim)f[x]=MP(0,bel[x]);
for(int i=0,t;i<vec[x].size();i++)
{
dfs1(t=vec[x][i]);
f[x]=min(f[x],AD(f[t],deep[bel[t]]-deep[bel[x]]));
}
}
PA up(int x,int y)
{
int ret=0;
for(int i=20;i>=0;i--)
if(y>>i&1)
ret+=sum[x][i],x=fa[x][i];
return MP(ret,x);
}
int move(int x,int y)
{
for(int i=20;i>=0;i--)
if(deep[fa[x][i]]>deep[y])
x=fa[x][i];
return x;
}
int cal(int x,int y)
{
PA v1=up(x,y),v2=up(fa[x][0],y);
return v2.first-v1.first;
}
void dfs2(int x)
{
ans[g[x].second]+=sum[bel[x]][0];
for(int i=0,t;i<vec[x].size();i++)
{
ans[g[x].second]-=sum[move(bel[t=vec[x][i]],bel[x])][0];
g[t]=min(f[t],AD(g[x],deep[bel[t]]-deep[bel[x]]));
dfs2(t);
int v1=g[t].first-g[x].first,v2=deep[bel[t]]-deep[bel[x]],p1,p2;
if(v1+v2<=1)p1=0;
else if((v1+v2)&1)p1=(v1+v2)/2;
else p1= g[x].second<g[t].second ? (v1+v2)/2 : (v1+v2)/2-1;
p1=max(p1,0);p1=min(p1,v2-1);p2=v2-1-p1;
int pos=up(bel[t],p2).second;
if(p1)ans[g[x].second]+=cal(pos,p1);
if(p2)ans[g[t].second]+=cal(bel[t],p2);
}
}
int main()
{
//freopen("tt.in","r",stdin);
scanf("%d",&n);
for(int i=1,x,y;i<n;i++)
{
scanf("%d%d",&x,&y);
add(x,y);add(y,x);
}
dfs(1,0);
for(int i=1;i<=20;i++)
for(int j=1;j<=n;j++)
{
fa[j][i]=fa[fa[j][i-1]][i-1];
sum[j][i]=sum[j][i-1]+sum[fa[j][i-1]][i-1];
}
scanf("%d",&Q);
for(int m;Q--;)
{
scanf("%d",&m);tim++;
for(int i=1;i<=m;i++)
{
scanf("%d",&a[i]),bj[a[i]]=tim;
b[i]=a[i];ans[b[i]]=0;
}
sort(a+1,a+1+m,cmp);
st[top=1]=1;vec[1].clear();
num[1]=1;bel[cnt=1]=1;
for(int i=1;i<=m;i++)
{
int t;
while((t=lca(st[top],a[i]))!=st[top])
{
if(deep[st[top-1]]<deep[t])
{
num[t]=++cnt;bel[cnt]=t;
vec[cnt].clear();
vec[cnt].push_back(num[st[top]]);
st[top]=t;
}
else vec[num[st[top-1]]].push_back(num[st[top]]),top--;
}
if(st[top]!=a[i])
{
st[++top]=a[i];
num[a[i]]=++cnt;bel[cnt]=a[i];
vec[cnt].clear();
}
}
for(int i=top;i>1;i--)
vec[num[st[i-1]]].push_back(num[st[i]]);
dfs1(1);
g[1]=f[1];dfs2(1);
for(int i=1;i<=m;i++)
printf("%d ",ans[b[i]]);
puts("");
}
return 0;
}