树链剖分模板——Luogu P3384

树链剖分有一棵线段树和多颗的版本

这里为了维护子树，用两遍dfs的一棵版本

发现好多讲解只有dfs和修改的操作

而没有如何将链biuld到线段树里的讲解

就好像你有云服务器和域名，但是不知道怎么解析DNS，有木有！

看下边：
树链是按照重儿子优先的dfs序存入线段树的
一个节点在dfs序中的顺序对应其在线段树上的区间节点顺序
换句话：在dfs序上建立线段树
然后用pos数组记录下每个节点在线段树上的位置（区间）相当于域名
那么biuld函数就相当于解析的过程啦
其他的楼下都有详解，
代码里也有大量讲解呦

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#define maxn 100010
#define ls l,m,rt<<1
#define rs m+1,r,rt<<1|1
#define LL long long 
using namespace std;
int a[maxn],n;
LL sum[maxn<<2],add[maxn<<2],bui=1;//bui建树用的计数器 
int h[maxn],v[maxn<<1],nt[maxn<<1],p=1;
int size[maxn],son[maxn],top[maxn],end[maxn],fa[maxn],dep[maxn],pos[maxn],dfsx[maxn],xu=1;
//size子节点数//son重儿子//top当前节点所属重链的头结点
//end[x]以x为根节点的子树在dfs序中最后一个节点的(dfs序中的)序号
//pos[x]x在dfs序中的位置,也是在线段树中的位置
//dfsx,dfs序.xu,计数器 
void add_edge(int x,int y){
    nt[p]=h[x];
    h[x]=p;
    v[p]=y;
    p++;
    nt[p]=h[y];
    h[y]=p;
    v[p]=x;
    p++;
}
void getup(int rt){
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void pushdown(int rt,int len){
    if (add[rt]!=0){
        add[rt<<1]+=add[rt];
        add[rt<<1|1]+=add[rt];
        sum[rt<<1]+=add[rt]*(len-(len>>1));
        sum[rt<<1|1]+=add[rt]*(len>>1);
        add[rt]=0;
    }
}
/******************************
建树是最重要的部分：
将树按(重儿子优先的)dfs序以链的形式存进线段树
并用pos记录其在线段树(dfs序中的位置)以便于修改查询
******************************/
void build(int l,int r,int rt){
    if (l==r){
        sum[rt]=a[dfsx[bui++]];
        return;
    }
    int m=(l+r)>>1;
    build(ls);
    build(rs);
    getup(rt);
}
void update(int l,int r,int rt,int L,int R,int k){
    if (L<=l&&r<=R){
        add[rt]+=k;
        sum[rt]+=(long long)k*(r-l+1);
        return ;
    }
    int m=(l+r)>>1;
    pushdown(rt,r-l+1);
    if (L<=m) update(ls,L,R,k);
    if (m+1<=R) update(rs,L,R,k);
    getup(rt);
}
LL query(int l,int r,int rt,int L,int R){
    if (L<=l&&r<=R){
        return sum[rt];
    } 
    pushdown(rt,r-l+1);
    LL tot=0;
    int m=(l+r)>>1;
    if (L<=m) tot+=query(ls,L,R);
    if (m+1<=R) tot+=query(rs,L,R);
    return tot;
}
void dfs1(int x){//处理fa,dep,size,son 
    int j=h[x],u=v[j],maxsize=0;//当u为叶子节点，son[u]=0 
    while(j){
        if (u!=fa[x]){
            dep[u]=dep[x]+1;
            fa[u]=x;//fa不要赋错值，不然段错误(递归死循环爆栈)
            size[x]++;
            dfs1(u);
            size[x]+=size[u];
            if (size[u]>maxsize){
                son[x]=u;
                maxsize=size[u];
            }
        }
        j=nt[j];
        u=v[j];
    }
    return;
}
int dfs2(int x,int start){//重链优先处理dfs序,top,end 
    bool f=1;//用来SPJ叶子节点及(一个节点有)多个叶子节点的情况 
    if (son[x]!=0){
        f=0;
        top[son[x]]=start;
        dfsx[xu++]=son[x];
        dfs2(son[x],start);
    }
    int j=h[x],u=v[j];
    while (j){
        if (u!=fa[x]&&u!=son[x]){
            f?top[u]=start,f=0:top[u]=u;//因为叶子节点不会再向下递归，所以下面就不用改了
            dfsx[xu++]=u;
            dfs2(u,u); 
        }
        j=nt[j];
        u=v[j];
    }
    end[x]=xu-1;
}
/***********************************
两点之间修改：
当不在同一链上时，修改top大的链，并跳到top[x]的父节点
跳至同一链之后，直接链上修改然后return即可
若为同一节点，修改后return
***********************************/
void update_path(int x,int y,int z){
    int fx=top[x],fy=top[y];
    while (fx!=fy){
        if (dep[fx]>=dep[fy]){
            update(1,n,1,pos[fx],pos[x],z);
            x=fa[fx];
        }else{
            update(1,n,1,pos[fy],pos[y],z);
            y=fa[fy];
        }       
        fx=top[x],fy=top[y];//这里不要忘了更新，不然死循环，其实直接用top可以避免
    }
    if (x!=y){//其实这里不用if仅保留下一行就好，不过为了警示还是写吧
        pos[x]<pos[y]?update(1,n,1,pos[x],pos[y],z):update(1,n,1,pos[y],pos[x],z);
    }else{
        update(1,n,1,pos[x],pos[y],z);
    }
}
LL query_path(int x,int y){
    LL ret=0;
    int fx=top[x],fy=top[y];
    while (fx!=fy){
        if (dep[fx]>=dep[fy]){
            ret+=query(1,n,1,pos[fx],pos[x]);
            x=fa[fx];
        }else{
            ret+=query(1,n,1,pos[fy],pos[y]);
            y=fa[fy];
        }
        fx=top[x],fy=top[y];        
    }
    if (x!=y){
        pos[x]<pos[y]?ret+=query(1,n,1,pos[x],pos[y]):ret+=query(1,n,1,pos[y],pos[x]);
    }else{
        ret+=query(1,n,1,pos[x],pos[y]);
    }
    return ret;
}
void update_subtree(int x,int z){
    update(1,n,1,pos[x],end[x],z);//因为一棵子树的dfs序是连续的，记录end后直接修改即可
}
LL query_subtree(int x){
    return query(1,n,1,pos[x],end[x]);
}
int main(){
    int m,r,x,y,z,Q;
    LL p;
    cin>>n>>m>>r>>p;
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    for (int i=1;i<n;i++) {
        scanf("%d%d",&x,&y);
        add_edge(x,y);
    }
    fa[r]=r;
    dep[r]=0;
    dfs1(r);
    top[r]=r;
    dfsx[xu++]=r;//初始化部分要看好
    dfs2(r,r);
    for (int i=1;i<xu;i++){
        pos[dfsx[i]]=i;
    }
    build(1,n,1);
    while (m--){
        scanf("%d",&Q);
        if (Q==1){
            scanf("%d%d%d",&x,&y,&z);
            update_path(x,y,z);
            continue;
        }
        if (Q==2){
            scanf("%d%d",&x,&y);
            printf("%lld\n",query_path(x,y)%p);
            continue;
        }
        if (Q==3){
            scanf("%d%d",&x,&z);
            update_subtree(x,z);
            continue;
        }
        if (Q==4){
            scanf("%d",&x);
            printf("%lld\n",query_subtree(x)%p);
            continue;
        }

    }
    return 0;
}