树链剖分入门

例题
POJ 3237()
BZOJ 3083
BZOJ 3531
BZOJ 3589
BZOJ 3626
将树上问题通过dfs序的性质转换为区间问题,从而对树上修改时就可转化为相应的区间修改。再通过引入重儿子,重链一系列概念将时间复杂度也变成了可以接受的层次

通过两次dfs求出树上一系列的信息

void dfs1(int u){
     sz[u]=1;
     dep[u]=dep[fa[u]]+1;

     for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].to;
        if(v==fa[u]) continue;
        fa[v]=u;
        dfs1(v);
        sz[u]+=sz[v];

        if(sz[son[u]]<sz[v]) son[u]=v;
     }
}

void dfs2(int u){

    static int T=0;
    seq[pos[u]=++T]=u;

    if(son[fa[u]]==u) top[u] = top[fa[u]];
    else top[u]=u;

    if(son[u]) dfs2(son[u]);

    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].to;
        if(v!=fa[u]&&v!=son[u]) dfs2(v);
    }
}

更新树上信息可分为 更新点权和更新边权
由于dfs序rk[]和pos[]数组是点的相应顺序排列的
所以当处理点权的时候我们直接更新两点间的信息即可
而当更新边权时,我们选择架空根节点,那么点权就可以变成边权了,所以对应跳重链的过程也有一点不同。具体看代码

点权

int QueryMax(int x,int y){
    int re=-inf;
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]])
            swap(x,y);
        re=max(re,treeQueryMax(1,1,n,pos[top[x]],pos[x]));
        x=fa[top[x]];
    }
    if(dep[x]<dep[y])
        swap(x,y);
    return re=max(re,treeQueryMax(1,1,n,pos[y],pos[x]));
}

边权
架空根节点后 dfs序的每个点的价值就变为该点与其父亲节点的边权值
那么在跳重链的过程中,假如dep[x]<dep[y],我们更新的其实应该是son[x]-y对应的边权(即点权)因为每个点x对应的权值是只向他的父亲的。
所以当两个点相同时就已经更新完成了

int QueryMax(int x,int y){
    int re=-inf;
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]])
            swap(x,y);
        re=max(re,treeQueryMax(1,1,n,pos[top[x]],pos[x]));
        x=fa[top[x]];
    }
    if(x==y) return ;
    if(dep[x]<dep[y]) swap(x,y);
    return re=max(re,treeQueryMax(1,1,n,pos[son[y]],pos[x]));
}
全部评论

相关推荐

10-13 17:47
门头沟学院 Java
wulala.god:图一那个善我面过,老板网上找的题库面的
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务