树上分块初步

最新推荐文章于 2019-07-20 01:23:06 发布

weixin_30475039

最新推荐文章于 2019-07-20 01:23:06 发布

阅读量552

点赞数

文章标签：数据结构与算法

原文链接：http://www.cnblogs.com/ZJXXCN/p/10713996.html

版权

转自这位大神

DFS序分块

将树节点按照DFS序每sqrtn

这种分块方式保证块的大小和数量，不保证块的连通和直径

Size分块

对于一个节点，如果它的父亲所属块的大小小于sqrtn

这种分块方式保证块的大小、直径，不保证块的数量

例题 BZOJ3720: Gty的妹子树

王室联邦分块

分块的名称来自BZOJ 1086: [SCOI2005]王室联邦

具体做法：

首先从根dfs整棵树，如果某棵子树剩余大小大于B，就将它单独成一块，并将它移除(即不再记录它的大小)
如果最后树的节点个数不足B，就把它加入上一块。z

这样分块每块大小为[B,3B)(只有一块，剩下的块直径不超过2B),直径不超过3B、但不保证连通。可以在树上莫队中使用

#include<cstdio>
#define maxn 1005
struct Edge{
    int next,to;
}edge[maxn*2];
int fi[maxn],se,n,b,stack[maxn],t,belong[maxn],size,head[maxn];
inline void add_edge(int u,int v){
    edge[++se].next=fi[u],edge[se].to=v,fi[u]=se,
    edge[++se].next=fi[v],edge[se].to=u,fi[v]=se;
}
void out(int s){
    size++;
    while(t>s)belong[stack[--t]]=size;
} 
void dfs(int x,int s,int fa){
    for(int i=fi[x];i;i=edge[i].next){
        int v=edge[i].to;
        if(v==fa)continue;
        dfs(v,t,x); 
        if(t-s>=b)out(s),head[size]=x;//如果当前子树栈中节点>=B,就自成一块 
    }
    stack[t++]=x;
}
int main(){
    scanf("%d%d",&n,&b);
    int u,v;
    for(int i=1;i<n;i++)scanf("%d%d",&u,&v),add_edge(u,v);
    dfs(1,0,0);
    while(t)belong[stack[--t]]=size;//将最后剩下的节点加入最后一个块 
    printf("%d\n",size);
    for(int i=1;i<=n;i++)printf("%d ",belong[i]);printf("\n");
    for(int i=1;i<=size;i++)printf("%d ",head[i]);
    return 0;
}

深度分块

设size=n−−√

这种分块方式可以保证块的个数、块的直径，但不能保证块的大小，用于解决对树上的链进行询问

int dfs(int x){//深度分块 
    int si=1;
    stack[top++]=x;//将x加入栈中 
    for(int i=fi[x];i;i=edge[i].next){
        int v=edge[i].to;
        if(v==fa[x])continue;
        dis[v]=edge[i].w,fa[v]=x,depth[v]=depth[x]+1,si+=dfs(v);
    }
    int k; 
    if(depth[x]%size==0&&si>=size){//如果当前点深度%size==0且子树大小>=size，则要和它的父亲分开 
        k=++s;block[x]=k;key[k]=x;//栈中x的子树中的节点和x一个块 
        while(stack[--top]!=x){
            block[stack[top]]=k;
        }
    }
    //否则和它的父亲一个块 
    return si;
}

例题 CodeChef TRIPS-Children Trips 树上分块， bzoj3351

保留块之间祖先关系的深度分块

设size=n−−√

这种分块可以保证块的直径和个数，不保证块的大小，可以解决以祖先关系为基础的询问

int dfs(int x,int depth){
    int si=1,ma=0;//ma记录他有几个儿子的子树有关键点 
    for(int i=fi[x];i;i=edge[i].next){
        int v=edge[i].to;
        fa[v]=x,si+=dfs(v,depth+1);
        if(k[v])ma++;//k==1表示v子树中存在关键点 
    }
    if((depth%size==0&&si>=size)||ma>1){ 
        k[x]=key[x]=1;//如果它本身是关键点 或 有2个及以上的儿子的子树有关键点，则它是关键点 
    }
    else if(ma)k[x]=1;//它的子树有关键点
    return si;
}
void dfs1(int x){
    if(key[x])block[x]=++s;
    for(int i=fi[x];i;i=edge[i].next){
        dfs1(edge[i].to);
    }
    int v=x;
    if(key[x]){
        v=x;while(!key[fa[v]])v=fa[v],block[v]=block[x];//将路径上的点加入当前块 
        v=fa[v];add_road(block[v],block[x]);//构建虚树中的边 
    }
    else if(!block[x]){
        while(!key[fa[v]])v=fa[v];
        v=fa[v],block[x]=-block[v];//不在路径上的点属于它最近关键点所在块(分开保存，所以存成负数)
    } 
}

转载于:https://www.cnblogs.com/ZJXXCN/p/10713996.html