luoguP3384 树链剖分模版

analysis

既然是模版题，按照模版打就行，只是注意开long long

树链剖分是解决在树上进行插点问线，插线问点等一系列树上的问题，一般体现为修改树上某条链（路径）上的点权值，查询树上某条链（路径）上的点权值

树链剖分可以理解为降维，将二维的树压成一维的数组

树链剖分就是把边哈希到线段树上的数据结构。

相当于在树上建立了重链组成的高铁，轻节点转移到重链上之后就可以搭高铁了（搭高铁的意思是，转移到重链之后，重链只需要查询（修改）一次就可以查询（修改）到结果）。

比如下面这个树和对应的数组如下：

我们不难得到：

• 重链上的点在数组中连续
• 对于节点x，其子树中的节点所在区间为[seg[x],seg[x]+siz[x]-1]
  利用这些性质即可完成本题所需操作

code

#include<bits/stdc++.h>
using namespace std;
#define loop(i,start,end) for(register int i=start;i<=end;++i)
#define anti_loop(i,start,end) for(register int i=start;i>=end;--i)
#define clean(arry,num) memset(arry,num,sizeof(arry))
#define max(a,b) ((a>b)?a:b)
#define min(a,b) ((a<b)?a:b)
#define ll long long
const int maxn=1e5+10;
const int maxm=1e5+10;
int n,m,root,mod,cnt=0;
int num[maxn];
int head[maxn];
struct node{int e;int nxt;}edge[maxn<<1];
int dep[maxn],son[maxn],siz[maxn],fa[maxn];//x的深度，重儿子，子树点个数，父节点
int top[maxn],seg[maxn],rev[maxn];//x的重链头，在线段树中的位置，rev[x]指线段树中位置x对应的点
ll sum[maxn<<2],lazy[maxn<<2];
template<typename T>void read(T &x){
    x=0;char r=getchar();T neg=1;
    while(r<'0'||r>'9'){if(r=='-')neg=-1;r=getchar();}
    while(r>='0'&&r<='9'){x=(x<<1)+(x<<3)+r-'0';r=getchar();}
    x*=neg;
}
inline void pushup(int rt){
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
inline void pushdown(int rt,int l,int r){
    if(lazy[rt]){
        lazy[rt<<1]+=lazy[rt],lazy[rt<<1|1]+=lazy[rt];
        sum[rt<<1]+=lazy[rt]*(((l+r)>>1)-l+1);
        //等价于lazy[rt]*(mid-l+1)
        sum[rt<<1|1]+=lazy[rt]*(r-(((l+r)>>1)+1)+1);
        //等价于lazy[rt]*(r-(mid+1)+1)
        //这里注意加括号，否则优先级不正确代价很大
        lazy[rt]=0;
    }
}
void build(int l,int r,int rt){
    if(l==r){
        sum[rt]=num[rev[l]];
        return;
    }
    int mid=((l+r)>>1);//注意大小于符号的方向！
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    pushup(rt);
}
ll query(int l,int r,int nl,int nr,int rt){
    if(l<=nl&&nr<=r){
        return sum[rt];
    }
    pushdown(rt,nl,nr);
    int mid=(nl+nr)>>1;
    ll _sum=0;
    if(mid>=l)_sum+=query(l,r,nl,mid,rt<<1);
    if(mid<r)_sum+=query(l,r,mid+1,nr,rt<<1|1);
    return _sum;
}
void update(int l,int r,int nl,int nr,int rt,int w){
    if(l<=nl&&nr<=r){
        lazy[rt]+=w;
        sum[rt]+=(nr-nl+1)*w;
        return;
    }
    pushdown(rt,nl,nr);
    int mid=(nl+nr)>>1;
    if(mid>=l)update(l,r,nl,mid,rt<<1,w);
    if(mid<r)update(l,r,mid+1,nr,rt<<1|1,w);
    pushup(rt);
}
void dfs2(int u,int f){//第二个dfs维护出seg，rev，top三个数组
    if(son[u]){//一定要先维护重儿子的标记，使得在线段树区间内重链连续排列
        seg[son[u]]=++seg[0];
        rev[seg[0]]=son[u];
        top[son[u]]=top[u];
        dfs2(son[u],u);
    }
    for(int i=head[u];i!=-1;i=edge[i].nxt){//维护完重儿子后再维护轻儿子
        int v=edge[i].e;
        if(!top[v]){//排除掉重儿子和父节点
            seg[v]=++seg[0];
            rev[seg[0]]=v;
            top[v]=v;
            dfs2(v,u);
        }
    }
}
void dfs1(int u,int f){//第一次dfs维护出fa，dep，siz，son四个数组
    fa[u]=f;
    dep[u]=dep[f]+1;
    siz[u]=1;
    son[u]=0;
    for(int i=head[u];i!=-1;i=edge[i].nxt){
        int v=edge[i].e;
        if(v==f)continue;
        dfs1(v,u);
        siz[u]+=siz[v];
        son[u]=((siz[v]>siz[son[u]])?v:son[u]);
    }
}
inline void addl(int u,int v){
    edge[cnt].e=v;
    edge[cnt].nxt=head[u];
    head[u]=cnt++;
}
inline void reply(){//回答函数
    loop(i,1,m){
        int op;read(op);
        if(op==1){
            int x,y,z;read(x),read(y),read(z);
            int fx=top[x],fy=top[y];
            while(fx!=fy){
                if(dep[fx]<dep[fy])swap(fx,fy),swap(x,y);
                update(seg[fx],seg[x],1,seg[0],1,z);
                x=fa[fx],fx=top[x];
            }
            if(dep[x]>dep[y])swap(x,y);
            update(seg[x],seg[y],1,seg[0],1,z);
        }
        else if(op==2){
            int x,y;read(x),read(y);ll res=0;
            int fx=top[x],fy=top[y];
            while(fx!=fy){
                if(dep[fx]<dep[fy])swap(fx,fy),swap(x,y);
                res+=query(seg[fx],seg[x],1,seg[0],1);res%=mod;
                x=fa[fx],fx=top[x];
            }
            if(dep[x]>dep[y])swap(x,y);
            res+=query(seg[x],seg[y],1,seg[0],1);
            printf("%lld\n",res%mod);
        }
        else if(op==3){
            int x,z;read(x),read(z);
            update(seg[x],seg[x]+siz[x]-1,1,seg[0],1,z);
        }
        else if(op==4){
            int x;read(x);
            printf("%lld\n",(query(seg[x],seg[x]+siz[x]-1,1,seg[0],1))%mod);
        }
    }
}
int main(){
    #ifndef ONLINE_JUDGE
    freopen("datain.txt","r",stdin);
    #endif
    clean(head,-1);clean(dep,0);clean(siz,0);clean(top,0);clean(sum,0);clean(lazy,0);
    read(n),read(m),read(root),read(mod);
    loop(i,1,n)read(num[i]);
    loop(i,1,n-1){
        int ui,vi;
        read(ui),read(vi);
        addl(ui,vi);
        addl(vi,ui);
    }
    dfs1(root,0);
    top[root]=root,seg[root]=1,rev[1]=root,seg[0]=1;
    //由于dfs2中无法对根节点赋值，因此要在dfs2以前将根的三个参数全部赋值完毕
    //seg[0]代表线段树中的点的个数
    dfs2(root,0);
    build(1,seg[0],1);
    reply();
    return 0;
}