洛谷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为根节点的子树内所有节点值之和

--by洛谷

---

---

一听名字就知道是模板；

有关树链剖分的内容

对子树操作，可理解为对dfs序上fa开头长度为子树size的区间操作；

代码如下：

 #include<cstdio>
 using namespace std;
 int n,m,r,L,R;
 int p,Z;
 int dis[];
 int ltree[];
 int      lz[];
 int dep[],fa[],hine[],size[];
 int top[],a[],rank[];
 struct ss{
     int to,next;
 }x[];
 int first[],num;
 void build(int ,int );
 void dfs_1(int );
 void dfs_2(int ,int );
 void up(int );
 void down(int ,int ,int );
 void builine(int ,int ,int );
 void swap(int&,int&);
 void work1(int );
 void work2(int );
 void add(int ,int ,int );
 int sum(int ,int ,int );
 int main()
 {
     int i,j,k;
     scanf("%d%d%d%d",&n,&m,&r,&p);
     for(i=;i<=n;i++)
         scanf("%d",&dis[i]),dis[i]%=p,hine[i]=i;
     for(i=;i<=n-;i++){
         scanf("%d%d",&j,&k);
         build(j,k);
         build(k,j);
     }
     dep[r]=;
     dfs_1(r);
     num=;
     dfs_2(r,r);
     num=;
     builine(,n,);
     for(i=;i<=m;i++){
         scanf("%d",&j);
         if(j<=)work1(j);
         else    work2(j-);
     }
 }
 void build(int f,int t){
     x[++num].next=first[f];
     x[num].to=t;
     first[f]=num;
 }
 void dfs_1(int now){
     int j=first[now];
     while(j){
         if(!dep[x[j].to]){
             dep[x[j].to]=dep[now]+;
             fa[x[j].to]=now;
             dfs_1(x[j].to);
             size[now]+=size[x[j].to];
             if(hine[now]==now||size[x[j].to]>size[hine[now]])
                 hine[now]=x[j].to;
         }
         j=x[j].next;
     }
     size[now]++;
 }
 void dfs_2(int now,int top_now){
     int j=first[now];
     top[now]=top_now;
     a[++num]=now;
     rank[now]=num;
     if(hine[now]!=now)
         dfs_2(hine[now],top_now);
     while(j){
         if(dep[x[j].to]==dep[now]+&&x[j].to!=hine[now])
             dfs_2(x[j].to,x[j].to);
         j=x[j].next;
     }
 }
 void up(int nu){
     ltree[nu]=ltree[nu<<]+ltree[nu<<|];
 }
 void down(int l,int r,int nu){
     if(!lz[nu])return ;
     int mid=(l+r)>>;
     lz[nu<<]=(lz[nu<<]+lz[nu])%p;
     lz[nu<<|]=(lz[nu<<|]+lz[nu])%p;
     ltree[nu<<]=(ltree[nu<<]+lz[nu]*(mid-l+))%p;
     ltree[nu<<|]=(ltree[nu<<|]+lz[nu]*(r-mid))%p;
     lz[nu]=;
 }
 void builine(int l,int r,int nu){
     if(l==r){
         ltree[nu]=dis[a[++num]];
         return;
     }
     int mid=(l+r)>>;
     builine(l,mid,nu<<);
     builine(mid+,r,nu<<|);
     up(nu);
 }
 void swap(int&a,int&b){
     int c=a;a=b;b=c;
 }
 void work1(int x){
     int u,v,ans=;
     scanf("%d%d",&u,&v);
     if(x==)scanf("%d",&Z);
     while(top[u]!=top[v]){
         if(dep[top[u]]>dep[top[v]])
             L=rank[top[u]],R=rank[u],u=fa[top[u]];
         else
             L=rank[top[v]],R=rank[v],v=fa[top[v]];
         if(x==)
             add(,n,);
         else
             ans=(ans+sum(,n,))%p;
     }
 //    if(u!=v){
     if(dep[u]>dep[v])
         swap(u,v);
     L=rank[u];R=rank[v];
     if(x==)
         add(,n,);
     else
         ans=(ans+sum(,n,))%p;
 //    }
     if(x==)
         printf("%d\n",ans);
 }
 void work2(int x){
     int ans=,i;
     scanf("%d",&i);
     L=rank[i];R=L+size[i]-;
     if(x==)scanf("%d",&Z);
         if(x==)
             add(,n,);
         else
             ans=(ans+sum(,n,))%p,printf("%d\n",ans);
 }
 void add(int l,int r,int nu){
     if(L<=l&&r<=R){
         ltree[nu]=(ltree[nu]+(r-l+)*Z)%p;
         lz[nu]=(lz[nu]+Z)%p;
         return ;
     }
     int mid=(l+r)>>;
     down(l,r,nu);
     if(L<=mid)
         add(l,mid,nu<<);
     if(R>mid)
         add(mid+,r,nu<<|);
     up(nu);
 }
 int sum(int l,int r,int nu){
     if(L<=l&&r<=R)
         return ltree[nu];
     int mid=(l+r)>>,ans=;
     down(l,r,nu);
     if(L<=mid)
         ans=(ans+sum(l,mid,nu<<))%p;
     if(R>mid)
         ans=(ans+sum(mid+,r,nu<<|))%p;
     return ans;
 }

祝AC哟！