树链剖分

树链剖分是一种对树进行划分的算法,将树分为多条链子,保证每个点只属于一条链,然后再通过数据结构(数组数组,BST,SPLAY,线段树等)来维护每一条链。

一般解决如下问题:

1.将树从x到y结点最短路径上所有的结点的数值都加上c

2.求树从x到y结点最短路径上所有节点的值之和

3.将以x为根节点的子树内所有节点值都加上z

4.求以x为根节点的子树内所有节点值之和

 

主要概念:

重儿子:在该节点的儿子中,子数数量最多的节点(如果存在多个最大的,选其一即可)(叶子节点无重儿子)

轻儿子:在该节点的儿子中,除重儿子之外的节点

重边:节点和其重儿子连成的边

轻边:节点和其轻儿子连成的边

重链:多条重边连成的路径

轻链:多条轻边连成的路径

 

 

 

如上图,1的重儿子是4,因为4有三棵子树。而2的重儿子是6。

而上图粗体黑线就是重链或者重边了。

 

基本的操作方法

1.在把所有的点连成树后我们先计算出每个点的父亲节点,深度,子树的数量,重儿子

2.完成第一步后我们将节点及其重儿子,重儿子的重儿子(重重儿子)等等连接成一条重链。通过这样的操作将树分成一条一条的重链

3.在重链上或者重链与重链之间进行具体的操作

 

1,2两步可以用两个dfs来完成

第一个dfs求出节点的子树数量(size),节点在树中的深度(deep),节点的父亲节点(fa),节点的重儿子(son)

int size[maxn],deep[maxn],fa[maxn],son[maxn];

void
dfs1(int u,int f,int dep){ // u代表节点,f代表父亲,dep代表深度 size[u]=1; deep[u]=dep; fa[u]=f; for(int i=head[u];i;i=tree[i].next){ int v=tree[i].to; //u的儿子 if(v==fa[u]) continue; dfs1(v,u,dep+1); //遍历儿子,遍历完之后就可以得到v的子树数量 size[u]+=size[tree[i].to]; if(size[tree[i].to]>size[son[u]]) //如果v的子树数量大于u原来重儿子的子树数量,v就是u的重儿子 son[u]=tree[i].to; } }

第一次dfs之后的节点情况如图

 

 

而第二次dfs则是求出每个节点所在的重链上的起始重儿子是谁,节点是第几次被dfs到的(dfs序),dfs序所对应的节点。

 

int tim,top[maxn],tid[maxn],rank[maxn];
//top 节点所在重链上的起始重儿子
//tid 节点的dfs序列
//rank dfs序对应的节点
 
void dfs2(int u,int t){ // u代表节点 ,t代表u所在重链的根部
    top[u]=t;
    tid[u]=++tim;
    rank[tim]=u;
    if(son[u]==-1) return ;
    else dfs2(son[u],t); //将该点的重儿子,重重儿子,重重重儿子等等连成一条重链
    for(int i=head[u];i;i=tree[i].next){
        int c=tree[i].to;
        if(v!=fa[u]&&v!=son[u]) //如果v不是u的重儿子,就构造新的重链 
        dfs2(v,v);
    } 
}

 

第二个dfs之后每个节点的情况

 

 

两次dfs之后根据第二次dfs求出的新编号建立线段树

void build(int l,int r,int rt){
    lazy[rt]=0;
    if(l==r){
        sum[rt]=a[rank[l]];  //根据dfs序来建树而不是 sum[rt]=a[l]; 
        return ;
    }
    int mid=(l+r)<<1;
    build(ls);
    build(rs);
    pushup(rt);
}

 

求解题目:

给定一棵树,改变x到y最短路径上的所有值。

思路:我们只需将两个节点移动到同一重链上并更新每次移动,在每次移动中都是移动起始重儿子深度更深的节点。

 

 

将x到节点y最短路径上所有节点的值增加z

假设x=7,y=13

先比较deep[top[7]]与deep[top[13]]的大小,7的更大,所以先更新区间(tid[top[7]],tid[7]),再令x=fa[top[7]],这时x就是节点1.

这时x=1,y=13,x和y便处于同一重链,最后再进行一次更新,更新区间(tid[1],tid[13])。

void update1(int x,int y,int c){
    while(top[x]!=top[y]){
        if(deep[top[x]]<deep[top[y]]) swap(x,y); //深度大的先修改
        update(tid[top[x]],tid[x],c,1,N,1);
        x=fa[top[x]]; 
    }
    if(deep[x]<deep[y]) swap(x,y);
    update(tid[y],tid[x],c,1,n,1);
}

 

而对于更新子树的操作,由第二次dfs可知以x为根节点的子树中,所有节点的dfs序是连续的,所以对子树的修改只需修改区间(tid[x],tid[x]+size[x]-1) 的值,也就是

update(tid[x],tid[x]+size[x]-1,c,1,n,1);

 

树链剖分有两个性质:

1.轻边(U,V),size(V)<=size(U)/2。

2.从根到某一点的路径上,不超过logN条轻边,不超过logN条重路径。

时间复杂度是O(nlog²n)。

洛谷的树链剖分模板题

题目描述

如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作:

操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z

操作2: 格式: 2 x y 表示求树从x到y结点最短路径上所有节点的值之和

操作3: 格式: 3 x z 表示将以x为根节点的子树内所有节点值都加上z

操作4: 格式: 4 x 表示求以x为根节点的子树内所有节点值之和

输入输出格式

输入格式:

第一行包含4个正整数N、M、R、P,分别表示树的结点个数、操作个数、根节点序号和取模数(即所有的输出结果均对此取模)。

接下来一行包含N个非负整数,分别依次表示各个节点上初始的数值。

接下来N-1行每行包含两个整数x、y,表示点x和点y之间连有一条边(保证无环且连通

接下来M行每行包含若干个正整数,每行表示一个操作,格式如下:

操作1: 1 x y z

操作2: 2 x y

操作3: 3 x z

操作4: 4 x

输出格式:

输出包含若干行,分别依次表示每个操作2或操作4所得的结果(对P取模)

输入输出样例

输入样例#1: 复制
5 5 2 24
7 3 7 8 0 
1 2
1 5
3 1
4 1
3 4 2
3 2 2
4 5
1 5 1 3
2 1 3
输出样例#1: 复制
2
21

说明

时空限制:1s,128M

数据规模:

对于30%的数据: N leq 10, M leq 10N10,M10

对于70%的数据: N leq {10}^3, M leq {10}^3N103,M103

对于100%的数据: N leq {10}^5, M leq {10}^5N105,M105

模板

 

#include<iostream>
#include<algorithm>
#include<cstring>
#include<bits/stdc++.h>
using namespace std;
const int maxn=100005;
#define ll long long int
#define ls l,mid,rt<<1
#define rs mid+1,r,rt<<1|1
int N,M,R,P,cnt,head[maxn<<1];
using namespace std;

struct edge{
    int to,next;
}e[maxn<<1];

void add(int x,int y){
    e[++cnt].to=y;
    e[cnt].next=head[x];
    head[x]=cnt;
}

int size[maxn],deep[maxn],fa[maxn],son[maxn];
void dfs1(int u,int f,int dep){
    size[u]=1;
    deep[u]=dep;
    fa[u]=f;
    for(int i=head[u];i;i=e[i].next){
        if(e[i].to==fa[u]) continue;
        dfs1(e[i].to,u,dep+1);
        size[u]+=size[e[i].to];
        if(size[e[i].to]>size[son[u]])
        son[u]=e[i].to;
    }
}

int tim,top[maxn],tid[maxn],rank[maxn];
void dfs2(int u,int t){
    top[u]=t;
    tid[u]=++tim;
    rank[tim]=u;
    if(son[u]==-1) return ;
    dfs2(son[u],t);
    for(int i=head[u];i;i=e[i].next){
        if(e[i].to!=fa[u]&&e[i].to!=son[u])
        dfs2(e[i].to,e[i].to);
    }
}

ll sum[maxn<<2],lazy[maxn<<2],a[maxn];

void pushup(int rt){
    sum[rt]=(sum[rt<<1]+sum[rt<<1|1])%P;
}

void build(int l,int r,int rt){
    if(l==r){
        sum[rt]=a[rank[l]]%P;
        return ;
    }
    int mid=(l+r)/2;
    build(ls);
    build(rs);
    pushup(rt);
}

void pushdown(int rt,int ln,int rn){
    if(lazy[rt]){
        lazy[rt<<1]=(lazy[rt<<1]+lazy[rt])%P;
        lazy[rt<<1|1]=(lazy[rt<<1|1]+lazy[rt])%P;
        sum[rt<<1]=(sum[rt<<1]+lazy[rt]*ln%P)%P;
        sum[rt<<1|1]=(sum[rt<<1|1]+lazy[rt]*rn%P)%P;
        lazy[rt]=0;
    }
}

void update(int L,int R,int c,int l,int r,int rt){
    if(L<=l&&R>=r){
        sum[rt]+=c*(r-l+1);
        lazy[rt]+=c;
        return ;
    }
    int mid=(l+r)/2;
    pushdown(rt,mid-l+1,r-mid);
    if(L<=mid) update(L,R,c,l,mid,rt<<1);
    if(R>mid) update(L,R,c,mid+1,r,rt<<1|1);
    pushup(rt);
}

ll query(int L,int R,int l,int r,int rt){
    if(L<=l&R>=r) return sum[rt];
    int mid=(l+r)/2;
    pushdown(rt,mid-l+1,r-mid);
    ll ans=0;
    if(L<=mid) ans=(ans+query(L,R,ls)%P)%P;
    if(R>mid) ans=(ans+query(L,R,rs)%P)%P;
    return ans%P;
}

void update1(int x,int y,int c){
    while(top[x]!=top[y]){
        if(deep[top[x]]<deep[top[y]]) swap(x,y);
        update(tid[top[x]],tid[x],c,1,N,1);
        x=fa[top[x]];
    }
    if(deep[x]<deep[y]) swap(x,y);
    update(tid[y],tid[x],c,1,N,1);
}

ll query1(int x,int y){
    ll ans=0;
    while(top[x]!=top[y]){
        if(deep[top[x]]<deep[top[y]]) swap(x,y);
        ans=(ans+query(tid[top[x]],tid[x],1,N,1)%P)%P;
        x=fa[top[x]];
    }
    if(deep[x]<deep[y]) swap(x,y);
    ans+=query(tid[y],tid[x],1,N,1)%P;
    return ans%P;
}

int main(){
    cin>>N>>M>>R>>P;
    for(int i=0;i<=N;i++){
        son[i]=-1;
    }
    for(int i=1;i<=N;i++)
    cin>>a[i];
    int x,y;
    for(int i=1;i<N;i++){
        cin>>x>>y;
        add(x,y);
        add(y,x);
    }
    dfs1(R,0,1);
    dfs2(R,R);
    build(1,N,1);
    int op,z;
    for(int i=1;i<=M;i++){
        cin>>op;
        if(op==1){
            cin>>x>>y>>z;
            update1(x,y,z);
        }
        if(op==2){
            cin>>x>>y;
            cout<<query1(x,y)<<endl;
        }
        if(op==3){
            cin>>x>>z;
            update(tid[x],tid[x]+size[x]-1,z,1,N,1);
        }
        if(op==4){
            cin>>x;
            cout<<query(tid[x],tid[x]+size[x]-1,1,N,1)%P<<endl;
        }
    }
    return 0;
}

 

原文地址:https://www.cnblogs.com/wushengyang/p/10808505.html