洛谷P3384 【模板】树链剖分

https://www.luogu.org/problemnew/show/P3384

已知一棵包含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为根节点的子树内所有节点值之和

存一下自己的板子

#include<bits/stdc++.h>
using namespace std;
#define maxn (100000+100)
#define ll long long

int n,m,root,mod,dfs_clock,op;
vector<int> G[maxn];
int fa[maxn],son[maxn],sz[maxn],deep[maxn],id[maxn],id2[maxn],top[maxn];
ll w[maxn],ver_w[maxn];

int ql,qr,val;
ll sumv[maxn*4],addv[maxn*4];
ll _sum;

int build(int o,int l,int r)
{
    int mid=(l+r)/2;
    if(l==r)return sumv[o]=addv[o]=w[l]%mod;
    return sumv[o]=(build(o*2,l,mid)+build(o*2+1,mid+1,r))%mod;
}

void maintain(int o,int l,int r)
{
    if(l==r)sumv[o]=addv[o]%mod;
    else sumv[o]=(sumv[o*2]+sumv[o*2+1]+(ll)addv[o]*(r-l+1))%mod;  
}

void query(int o,int l,int r,ll add)
{
    if(ql<=l&&qr>=r)_sum=(_sum+add*(r-l+1)+sumv[o])%mod;
    else
    {
        int mid=(l+r)/2;
        if(ql<=mid)query(o*2,l,mid,(add+addv[o])%mod);
        if(qr>mid)query(o*2+1,mid+1,r,(add+addv[o])%mod);
    }  
}

void update(int o,int l,int r)
{
    if(ql<=l&&qr>=r)addv[o]=(addv[o]+val)%mod;
    else
    {
        int mid=(l+r)/2;
        if(ql<=mid)update(o*2,l,mid);
        if(qr>mid)update(o*2+1,mid+1,r);
    }
    maintain(o,l,r);
}

void dfs1(int u,int f)
{
    deep[u]=deep[f]+1;
    sz[u]=1;
    fa[u]=f;
    int maxx=0;
    for(int i=0;i<G[u].size();i++)
    {
        int v=G[u][i];
        if(v==f)continue;
        dfs1(v,u);
        sz[u]+=sz[v];
        if(sz[v]>maxx){maxx=sz[v];son[u]=v;}
    }
}

void dfs2(int u,int up)
{
    id[u]=id2[u]=++dfs_clock;
    top[u]=up;
    if(son[u]){dfs2(son[u],up);id2[u]=id2[son[u]];}
    for(int i=0;i<G[u].size();i++)
    {
        int v=G[u][i];
        if(v==fa[u] || v==son[u])continue;
        dfs2(v,v);
        id2[u]=id2[v];
    }
}

void Update1(int u,int v,int z)
{
    int tpu=top[u],tpv=top[v];
    while(tpu!=tpv)
    {
        if(deep[tpu]<deep[tpv]){swap(tpu,tpv);swap(u,v);}
        ql=id[tpu],qr=id[u],val=z;
        update(1,1,n);
        u=fa[tpu];
        tpu=top[u];
    }
    if(deep[u]>deep[v])swap(u,v);
    ql=id[u],qr=id[v],val=z;
    update(1,1,n);
}

ll Query1(int u,int v)
{
    _sum=0;
    int tpu=top[u],tpv=top[v];
    while(tpu!=tpv)
    {
        if(deep[tpu]<deep[tpv]){swap(tpu,tpv);swap(u,v);}
        ql=id[tpu],qr=id[u];
        query(1,1,n,0);
        u=fa[tpu];
        tpu=top[u];
    }
    if(deep[u]>deep[v])swap(u,v);
    ql=id[u],qr=id[v];
    query(1,1,n,0);
    return _sum;
}

void Update2(int u,int z)
{
    ql=id[u],qr=id2[u],val=z;
    update(1,1,n);
}

ll Query2(int u)
{
    _sum=0;
    ql=id[u],qr=id2[u];
    query(1,1,n,0);
    return _sum;
}

int main()
{
    //freopen("input.in","r",stdin);
    cin>>n>>m>>root>>mod;
    int x,y,z;
    for(int i=1;i<=n;i++)scanf("%d",&ver_w[i]);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        G[x].push_back(y);
        G[y].push_back(x);
    } 
    dfs1(root,0);
    dfs2(root,root);
    for(int i=1;i<=n;i++)w[id[i]]=ver_w[i];
    build(1,1,n);  
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&op);
        switch(op)
        {
            case 1:
                scanf("%d%d%d",&x,&y,&z);
                Update1(x,y,z);
                break;
            case 2:
                scanf("%d%d",&x,&y);
                printf("%lld\n",Query1(x,y));
                break;
            case 3:
                scanf("%d%d",&x,&z);
                Update2(x,z);
                break;
            case 4:
                scanf("%d",&x);
                printf("%lld\n",Query2(x));
                break;
        }
    }
    return 0;
}