【Hnoi2014】世界树

时间限制 : 2000 MS 空间限制 : 565536 KB

问题描述

世界树是一棵无比巨大的树，它伸出的枝干构成了整个世界。在这里，生存着各种各样的种族和生灵，他们共同信奉着绝对公正公平的女神艾莉森，在他们的信条里，公平是使世界树能够生生不息、持续运转的根本基石。
世界树的形态可以用一个数学模型来描述：世界树中有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-l行，每行两个正整数x，y，表示x聚居地与y聚居地之间有一条长度为1的双向道路。接下来一行为一个正整数q，表示国王询问的年数。 接下来q块，每块两行：
第i块的第一行为1个正整数m[i]，表示第i年授权的临时议事处的个数。
第i块的第二行为m[i]个正整数h[l]、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

题解

虚树类题目的一道经典题。按套路构建虚树，关键的地方在于如何在虚树上进行DP讨论……
本蒟蒻最开始的想法是分关键点与非关键点两种情况讨论，对每种情况为又要分子节点是否是关键点两种子情况，而且讨论过程十分恶心，写到一半把自己绕晕了，于是求教了各大佬的博客，发现正解太妙了，不知道比我高到哪里去了……
首先我们要求出每个非关键点从属于那个关键点，这需要讨论到每个节点与其父亲及儿子的关系（最短距离）。这种情况需要两次DP讨论：一次讨论与儿子的关系，先讨论儿子节点再由儿子节点的状态转移到自己；另一次讨论与父亲的关系，DP时先由自己更新儿子状态再讨论儿子。
然后我们再求出每个关键点包含的虚树外的点有哪些，这只需要一遍树DP讨论即可。如果两个相邻点所属点相同，则直接加上路径上的点数；如果不同，则要把路径上的点分割，寻找分割点的方法类比求LCA时的逼近法即可。

代码

#include<stdio.h>
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cmath>
using namespace std;
const int maxn=3e5+5;
const int inf=1e9;
int a,b,n,m,q,vt,cnt1,dfn[maxn],Last1[maxn];
int dep[maxn],siz[maxn],fa[maxn][25];
int top,cnt2,f[maxn],g[maxn],h[maxn],s[maxn],bl[maxn],ans[maxn],Last2[maxn];
struct node{
    int End,Next,Len;
}edge1[maxn*2],edge2[maxn*2];
bool cmp(int x,int y)
{
    return dfn[x]<dfn[y];
}
void save(int x,int y)
{
    edge1[++cnt1].End=y;
    edge1[cnt1].Next=Last1[x],Last1[x]=cnt1;
}
void add(int x,int y,int z)
{
    edge2[++cnt2].End=y,edge2[cnt2].Len=z;
    edge2[cnt2].Next=Last2[x],Last2[x]=cnt2;
}
void DFS(int x)
{
    dfn[x]=++vt,siz[x]=1;
    dep[x]=dep[fa[x][0]]+1;
    int k=ceil(log2(dep[x]));
    for(int i=1;i<=k;i++) fa[x][i]=fa[fa[x][i-1]][i-1];
    for(int i=Last1[x];i;i=edge1[i].Next)
    {
        int y=edge1[i].End;
        if(y==fa[x][0]) continue;
        fa[y][0]=x,DFS(y),siz[x]+=siz[y];
    }
}
int getlca(int x,int y)
{
    if(dep[x]<dep[y]) swap(x,y);
    int k=dep[x]-dep[y],s=ceil(log2(n));
    for(int i=0;i<=s;i++)
        if(k&(1<<i)) x=fa[x][i];
    if(x==y) return x;
    k=ceil(log2(dep[x]));
    for(int i=k;i>=0;i--)
        if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
void build_tree()
{
    s[++top]=1;
    for(int i=1;i<=m;i++)
    {
        int lca=getlca(h[i],s[top]);
        while(dep[s[top-1]]>dep[lca]) add(s[top-1],s[top],dep[s[top]]-dep[s[top-1]]),top--;
        if(s[top]!=lca) add(lca,s[top],dep[s[top]]-dep[lca]),top--;
        if(s[top]!=lca) s[++top]=lca;
        if(s[top]!=h[i]) s[++top]=h[i];
    }
    while(--top) add(s[top],s[top+1],dep[s[top+1]]-dep[s[top]]);
}
int getup(int x,int l)
{
    int s=ceil(log2(n));
    for(int i=0;i<=s;i++)
        if(l&(1<<i)) x=fa[x][i];
    return x;
}
int getmid(int x,int y,int l)
{
    if(l&1)
        {
            int mid=getup(y,l/2-f[y]);
            ans[bl[y]]+=siz[mid]-siz[y],ans[bl[x]]-=siz[mid];
        }
    else
    {
        if(bl[x]<bl[y])
        {
            int mid=getup(y,l/2-1-f[y]);
            ans[bl[y]]+=siz[mid]-siz[y],ans[bl[x]]-=siz[mid];
        }
        else
        {
            int mid=getup(y,l/2-f[y]);
            ans[bl[y]]+=siz[mid]-siz[y],ans[bl[x]]-=siz[mid];
        }
    }
}
void getbl1(int x)
{ 
    for(int i=Last2[x];i;i=edge2[i].Next)
    {
        int y=edge2[i].End;
        getbl1(y);
        if(!bl[x]) bl[x]=bl[y],f[x]=f[y]+edge2[i].Len;
        else
        {
            if(f[x]>f[y]+edge2[i].Len) bl[x]=bl[y],f[x]=f[y]+edge2[i].Len;
            else if(f[x]==f[y]+edge2[i].Len&&bl[y]<bl[x]) bl[x]=bl[y],f[x]=f[y]+edge2[i].Len;
        }
    }
}
void getbl2(int x)
{
    for(int i=Last2[x];i;i=edge2[i].Next)
    {
        int y=edge2[i].End;
        if(!bl[y]) bl[y]=bl[x],f[y]=f[x]+edge2[i].Len;
        else
        {
            if(f[y]>f[x]+edge2[i].Len) bl[y]=bl[x],f[y]=f[x]+edge2[i].Len;
            else if(f[y]==f[x]+edge2[i].Len&&bl[x]<bl[y]) bl[y]=bl[x],f[y]=f[x]+edge2[i].Len;
        }
        getbl2(y);
    }
}
void getans(int x)
{
    ans[bl[x]]+=siz[x];
    for(int i=Last2[x];i;i=edge2[i].Next)
    {
        int y=edge2[i].End;
        if(bl[x]!=bl[y]) getmid(x,y,edge2[i].Len+f[x]+f[y]);
        else ans[bl[x]]-=siz[y];
        getans(y);
    }
    Last2[x]=f[x]=bl[x]=0;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&a,&b);
        save(a,b),save(b,a);
    }
    DFS(1);
    scanf("%d",&q);
    while(q--)
    {
        cnt2=top=0;
        scanf("%d",&m);
        for(int i=1;i<=m;i++) scanf("%d",&h[i]),g[i]=bl[h[i]]=h[i];
        sort(h+1,h+1+m,cmp);
        build_tree(),getbl1(1),getbl2(1),getans(1);
        for(int i=1;i<=m;i++) printf("%d ",ans[g[i]]),ans[g[i]]=0;
        puts("");
    }
    return 0;
}