世界树

世界树

时间限制: 2 Sec 内存限制: 512 MB

题目描述
世界树是一棵无比巨大的树，它伸出的枝干构成了整个世界。在这里，生存着各种各样的种族和生灵，他们共同信奉着绝对公正公平的女神艾莉森，在他们的信条里，公平是使世界树能够生生不息、持续运转的根本基石。
世界树的形态可以用一个数学模型来描述： 世界树中有n 个种族，种族的编号分别从1到n，分别生活在编号为1 到n 的聚居地上，种族的编号与其聚居地的编号相同。有的聚居地之间有双向的道路相连，道路的长度为1。保证连接的方式会形成一棵树结构，即所有的聚居地之间可以互相到达，并且不会出现环。定义两个聚居地之间的距离为连接他们的道路的长度；
例如，若聚居地a 和b 之间有道路，b 和c 之间有道路，因为每条道路长度为1 而且又不可能出现环，所以a 与c 之间的距离为2。
出于对公平的考虑，第i 年，世界树的国王需要授权m[i]个种族的聚居地为临时议事处。
对于某个种族x（x 为种族的编号），如果距离该种族最近的临时议事处为y（y 为议事处所在聚居地的编号），则种族x 将接受y 议事处的管辖（如果有多个临时议事处到该聚居地的距离一样，则y 为其中编号最小的临时议事处）。
现在国王想知道，在q 年的时间里，每一年完成授权后，当年每个临时议事处将会管理多少个种族（议事处所在的聚居地也将接受该议事处管理）。
现在这个任务交给了以智慧著称的灵长类的你：程序猿。请帮国王完成这个任务吧。

输入
第一行为一个正整数n，表示世界树中种族的个数。
接下来n-1 行，每行两个正整数x,y，表示x 聚居地与y 聚居地之间有一条长度为1 的双向道路。
接下来一行为一个正整数q，表示国王询问的年数。
接下来q 块，每块两行：
第i 块的第一行为1 个正整数m[i]，表示第i 年授权的临时议事处的个数。
第i 块的第二行为m[i]个正整数h[1]、h[2]、…、h[m[i]]，表示被授权为临时议事处的聚居地编号（保证互不相同）。

输出
输出包含q 行，第i 行为m[i]个整数，该行的第j（j=1,2,…,m[i]）个数表示第i 年被授权的聚居地h[j]的临时议事处管理的种族个数。

样例输入
10
2 1
3 2
4 3
5 4
6 1
7 3
8 3
9 4
10 1
5
2
6 1
5
2 7 3 6 9
1
8
4
8 7 10 3
5
2 9 3 5 8

样例输出
1 9
3 1 4 1 1
10
1 1 3 5
4 1 3 1 1

数据范围
n≤300000, q≤300000, m[1]+m[2]+…+m[q] ≤300000

题解

很容易想到将关键点建虚树搞。
可以先求出虚数上每个点分别被哪个关键点控制。
考虑虚树上每一条边，如果两个点都被控制那么直接统计答案。
否则可以在这条边上倍增，寻找分界点。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
#define inf 1050000000
#define N 300010
#define D 19
using namespace std;
int n,m,Q,cnt,s[N],h[N],t[N],po[N],ans[N],rem[N];
int dep[N],bel[N],fa[N],w[N][D+1],size[N];
int k,la[N],ff[N],k1,lx[N],fx[N*2],l[N],r[N];
struct node{int a,b;}e[N*2],map[N];
bool cmp(const int &x,const int &y){return l[x]<l[y];}

void adde(int a,int b)
{
  e[++k1]=(node){a,b};fx[k1]=lx[a];lx[a]=k1;
  e[++k1]=(node){b,a};fx[k1]=lx[b];lx[b]=k1;
}

void add(int a,int b)
{
  map[++k]=(node){a,b};
  ff[k]=la[a];la[a]=k;
}

void dfs(int x)
{
  l[x]=++cnt;size[x]=1;
  for(int i=1;i<=D;i++)
    w[x][i]=w[w[x][i-1]][i-1];
  for(int a=lx[x];a;a=fx[a])
    if(w[x][0]!=e[a].b)
    {
      dep[e[a].b]=dep[x]+1;
      w[e[a].b][0]=x;dfs(e[a].b);
      size[x]+=size[e[a].b];
    }
  r[x]=cnt;
}

int lca(int a,int b)
{
  if(dep[a]<dep[b])swap(a,b);
  for(int i=D;i>=0;i--)
    if(dep[w[a][i]]>=dep[b])a=w[a][i];
  if(a==b)return a;
  for(int i=D;i>=0;i--)
    if(w[a][i]!=w[b][i])a=w[a][i],b=w[b][i];
  return w[a][0];
}

int dis(int a,int b)
{
  if(!a||!b)return inf;
  return dep[a]+dep[b]-dep[lca(a,b)]*2;
}

void dfs1(int x)
{
  rem[x]=size[x];
  for(int a=la[x];a;a=ff[a])
  {
    dfs1(map[a].b);
    if(!bel[map[a].b])continue;
    int l1=dis(bel[map[a].b],x);
    int l2=dis(bel[x],x);
    if(l1<l2||(l1==l2&&bel[map[a].b]<bel[x]))
      bel[x]=bel[map[a].b];
  }
}

void dfs2(int x)
{
  for(int a=la[x];a;a=ff[a])
  {
    int l1=dis(bel[x],map[a].b);
    int l2=dis(bel[map[a].b],map[a].b);
    if(l1<l2||(l1==l2&&bel[x]<bel[map[a].b]))
      bel[map[a].b]=bel[x];
    dfs2(map[a].b);
  }
}

int find(int x,int des)
{
  for(int i=D;i>=0;i--)
    if(dep[w[x][i]]>=des)x=w[x][i];
  return x;
}

void solve()
{
  int top=1,tot=m,st=2;t[1]=1;k=0;
  if(s[1]>1)po[++tot]=1,st=1;
  for(int i=1;i<=m;i++)
    po[i]=s[i],bel[s[i]]=s[i],ans[s[i]]=0;
  for(int i=st;i<=m;i++)
  {
    int pos=lca(s[i],t[top]);
    while(dep[t[top]]>dep[pos])
    {
      if(dep[t[top-1]]<=dep[pos])fa[t[top]]=pos;
      top--;
    }
    if(t[top]!=pos)fa[pos]=t[top],po[++tot]=pos,t[++top]=pos;
    fa[s[i]]=pos;t[++top]=s[i];
  }
  for(int i=1;i<=tot;i++)la[po[i]]=0;
  for(int i=1;i<=tot;i++)add(fa[po[i]],po[i]);
  dfs1(1);dfs2(1);
  for(int i=1;i<=tot;i++)
    for(int a=la[po[i]];a;a=ff[a])
    {
      int x=po[i],y=map[a].b,A=bel[x],B=bel[y];
      int pos=find(y,dep[x]+1);rem[x]-=size[pos];
      if(A==B){ans[A]+=size[pos]-size[y];continue;}
      int mid=y;
      for(int i=D;i>=0;i--)
      {
        int des=w[mid][i];
        if(dep[des]<=dep[x])continue;
        int l1=dis(A,des),l2=dis(des,B);
        if(l1>l2||(l1==l2&&A>B))mid=des;
      }
      ans[A]+=size[pos]-size[mid];
      ans[B]+=size[mid]-size[y];
    }
  for(int i=1;i<=tot;i++)ans[bel[po[i]]]+=rem[po[i]];
  for(int i=1;i<=m;i++)printf("%d ",ans[h[i]]);
  printf("\n");
  for(int i=1;i<=tot;i++)bel[po[i]]=rem[po[i]]=0;
}

int main()
{
  int a,b;
  scanf("%d",&n);
  for(int i=1;i<n;i++)
    scanf("%d%d",&a,&b),adde(a,b);
  dep[1]=1;dfs(1);
  scanf("%d",&Q);
  while(Q--)
  {
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
      scanf("%d",&s[i]),h[i]=s[i];
    sort(s+1,s+m+1,cmp);solve();
  }
  return 0;
}