树链剖分 模板

洛谷：P3384 【模板】树链剖分

#include<cstdio>
#include<iostream>
#include<algorithm>
#define ls (k*2)
#define rs (k*2+1)
#define down() {lz[ls]+=lz[k];lz[rs]+=lz[k];(sum[ls]+=(mid-ll+1)*lz[k])%=mod;(sum[rs]+=(rr-mid)*lz[k])%=mod;lz[k]=0;} //注意是ll和rr，不是L和R，又调了半天。。。
using namespace std;
const int MAXN=100099,MAXM=200099;
int n,m,root,mod,num,DFN,op,x,y,z;
int head[MAXN],from[MAXM],to[MAXM],Next[MAXM];
int top[MAXN],w1[MAXN],w2[MAXN],fa[MAXN],son[MAXN],Val[MAXN],dep[MAXN],sz[MAXN],lz[MAXN*5],sum[MAXN*5];
void add(int f,int t)
{
  Next[++num]=head[f];
  to[num]=t;
  head[f]=num;
}
void dfs1(int u,int FA)
{
  sz[u]=1;
  for(int i=head[u],v=to[i];i;i=Next[i],v=to[i])
    if(v!=FA)
      {
        dep[v]=dep[u]+1;
        fa[v]=u;
        dfs1(v,u);
        sz[u]+=sz[v];
        if(sz[v]>sz[son[u]]) son[u]=v;
      }
}
void dfs2(int u)
{
  w1[u]=++DFN;
  if(son[u]!=0)
    {
      top[son[u]]=top[u];
      dfs2(son[u]);
    }
  for(int i=head[u],v=to[i];i;i=Next[i],v=to[i])
    if(v!=son[u] && !w1[v])
      {
        top[v]=v;
        dfs2(v);
      }
  w2[u]=DFN;
}
void Update(int k,int ll,int rr,int L,int R,int V)
{
  if(ll==L && rr==R){sum[k]+=(R-L+1)*V%mod; sum[k]%=mod; lz[k]+=V; return;}
  int mid=(ll+rr)/2; down();
  if(R<=mid) Update(ls,ll,mid,L,R,V);
  else if(mid<L) Update(rs,mid+1,rr,L,R,V);
  else Update(ls,ll,mid,L,mid,V) , Update(rs,mid+1,rr,mid+1,R,V);
  sum[k]=(sum[ls]+sum[rs])%mod;
}
int Query(int k,int ll,int rr,int L,int R)
{
  if(ll==L && rr==R){return sum[k];}
  int mid=(ll+rr)/2; down();
  if(R<=mid) return Query(ls,ll,mid,L,R);
  else if(mid<L) return Query(rs,mid+1,rr,L,R);
  else return (Query(ls,ll,mid,L,mid)+Query(rs,mid+1,rr,mid+1,R))%mod;
}
int main()
{
  scanf("%d%d%d%d",&n,&m,&root,&mod);
  for(int i=1;i<=n;i++) scanf("%d",&Val[i]);
  for(int i=1,a,b;i<n;i++)
    {
      scanf("%d%d",&a,&b);
      add(a,b); add(b,a);
    }
  top[root]=root;
  dep[root]=1;
  dfs1(root,0);
  dfs2(root);
  for(int i=1;i<=n;i++) Update(1,1,n,w1[i],w1[i],Val[i]%mod);
  while(m--)
    {
      scanf("%d",&op);
      if(op==1)
        {
          scanf("%d%d%d",&x,&y,&z); z%=mod;
          while(top[x]!=top[y])
            {
              if(dep[top[x]]<dep[top[y]]) swap(x,y);
              Update(1,1,n,w1[top[x]],w1[x],z);
              x=fa[top[x]];
            }
          if(dep[x]<dep[y]) swap(x,y);
          Update(1,1,n,w1[y],w1[x],z);
        }
      else if(op==2)
        {
          scanf("%d%d",&x,&y);
          int ans=0;
          while(top[x]!=top[y])
            {
              if(dep[top[x]]<dep[top[y]]) swap(x,y);
              (ans+=Query(1,1,n,w1[top[x]],w1[x]))%=mod;
              x=fa[top[x]];
            }
          if(dep[x]<dep[y]) swap(x,y);
          (ans+=Query(1,1,n,w1[y],w1[x]))%=mod;
          printf("%d
",ans%mod);
        }
      else if(op==3)
        {
          scanf("%d%d",&x,&z); z%=mod;
          Update(1,1,n,w1[x],w2[x],z);
        }
      else if(op==4)
        {
          scanf("%d",&x);
          printf("%d
",Query(1,1,n,w1[x],w2[x]));
        }
    }
  return 0;
}