【bzoj3572】[Hnoi2014]世界树 虚树+倍增lca+DP

4 篇文章 0 订阅
2 篇文章 0 订阅

Description

世界树是一棵无比巨大的树,它伸出的枝干构成了整个世界。在这里,生存着各种各样的种族和生灵,他们共同信奉着绝对公正公平的女神艾莉森,在他们的信条里,公平是使世界树能够生生不息、持续运转的根本基石。
世界树的形态可以用一个数学模型来描述:世界树中有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年的时间里,每一年完成授权后,当年每个临时议事处将会管理多少个种族(议事处所在的聚居地也将接受该议事处管理)。 现在这个任务交给了以智慧著称的灵长类的你:程序猿。请帮国王完成这个任务吧。

Input

第一行为一个正整数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]],表示被授权为临时议事处的聚居地编号(保证互不相同)。

Output

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

Sample Input

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
Sample Output

1 9

3 1 4 1 1

10

1 1 3 5

4 1 3 1 1
HINT

N<=300000, q<=300000,m[1]+m[2]+…+m[q]<=300000

题解
https://www.cnblogs.com/zzqsblog/p/5560645.html
zzq dalao已经讲的很详尽了。

代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<map>
#define ll long long
#define inf 1000000000
#define mod 1000000007
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
const int N=300005;
int tot,top,n,m,Q,dfn,q[N];
int Head[N],ret[N<<1],Next[N<<1];
int dep[N],fa[N][19],size[N];
int ans[N],remain[N],bel[N];
int a[N],b[N],c[N],id[N],bin[N];
inline bool cmp(int a,int b){return id[a]<id[b];}
inline int lca(int x,int y)
{
    if(dep[x]>dep[y])swap(x,y);
    int t=dep[y]-dep[x];
    for(int i=0;bin[i]<=t;i++)
        if(bin[i]&t)y=fa[y][i];
    for(int i=18;i>=0;i--)
        if(fa[x][i]!=fa[y][i])
            x=fa[x][i],y=fa[y][i];
    return x==y?x:fa[x][0];
}
inline int dis(int a,int b)
{
    return dep[a]+dep[b]-2*dep[lca(a,b)];
}
inline void ins(int u,int v){ret[++tot]=v;Next[tot]=Head[u];Head[u]=tot;}
void dfs(int u,int f)
{
    id[u]=++dfn;size[u]=1;
    for (int i=1;bin[i]<=dep[u];i++) fa[u][i]=fa[fa[u][i-1]][i-1]; 
    for (int i=Head[u];i;i=Next[i])
    {
        if (ret[i]==f) continue;
        fa[ret[i]][0]=u;dep[ret[i]]=dep[u]+1;
        dfs(ret[i],u);size[u]+=size[ret[i]];
    }
}
void dfs1(int u)
{
    c[++dfn]=u;remain[u]=size[u];
    for (int i=Head[u];i;i=Next[i])
    {
        dfs1(ret[i]);if (!bel[ret[i]]) continue;
        int t1=dis(bel[ret[i]],u),t2=dis(bel[u],u);
        if ((t1==t2&&bel[ret[i]]<bel[u])||t1<t2||!bel[u])bel[u]=bel[ret[i]];
    }
}
void dfs2(int u)
{
    for (int i=Head[u];i;i=Next[i])
    {
        int t1=dis(bel[u],ret[i]),t2=dis(bel[ret[i]],ret[i]);
        if((t1==t2&&bel[ret[i]]>bel[u])||t1<t2||!bel[ret[i]])
            bel[ret[i]]=bel[u];
        dfs2(ret[i]);
    }
}
inline void solve(int a,int b)
{
    int x=b,mid=b;
    for (int i=18;i>=0;i--)
        if (dep[fa[x][i]]>dep[a])x=fa[x][i];
    remain[a]-=size[x];
    if (bel[a]==bel[b])
    {
        ans[bel[a]]+=size[x]-size[b];
        return;
    }
    for (int i=18;i>=0;i--)
    {
        int x=fa[mid][i];
        if (dep[x]<=dep[a]) continue;
        int t1=dis(bel[a],x),t2=dis(bel[b],x);
        if (t1>t2||(t1==t2&&bel[a]>bel[b])) mid=x;
    }
    ans[bel[a]]+=size[x]-size[mid];
    ans[bel[b]]+=size[mid]-size[b];
}
inline void query()
{
    m=read();top=dfn=tot=0;
    for (int i=1;i<=m;i++) a[i]=b[i]=read(),bel[a[i]]=a[i];
    sort(a+1,a+m+1,cmp);if(bel[1]!=1) q[++top]=1;
    for (int i=1;i<=m;i++)
    {
        int t=a[i],f=0;
        while (top>0)
        {
            f=lca(t,q[top]);
            if (top>1&&dep[f]<dep[q[top-1]])ins(q[top-1],q[top]),--top;
            else if (dep[f]<dep[q[top]]){ins(f,q[top]);top--;break;}
            else break;
        }
        if (q[top]!=f) q[++top]=f;
        q[++top]=t;
    }
    while (top>1) ins(q[top-1],q[top]),top--;
    dfs1(1);dfs2(1);
    for (int i=1;i<=dfn;i++)
        for (int j=Head[c[i]];j;j=Next[j]) solve(c[i],ret[j]);
    for (int i=1;i<=dfn;i++) ans[bel[c[i]]]+=remain[c[i]];
    for (int i=1;i<=m;i++) printf("%d ",ans[b[i]]);puts("");
    for (int i=1;i<=dfn;i++) ans[c[i]]=Head[c[i]]=bel[c[i]]=remain[c[i]]=0;
}
int main()
{
    bin[0]=1;for(int i=1;i<20;i++)bin[i]=bin[i-1]<<1;
    n=read();
    for (int i=1;i<n;i++)
    {
        int u=read(),v=read();
        ins(u,v);ins(v,u);
    }
    dfs(1,0);
    Q=read();for (int i=1;i<=n;i++) Head[i]=0;
    while (Q--)query();
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值