树链剖分 模板

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

  1 #include<cstdio>
  2 #include<iostream>
  3 #include<algorithm>
  4 #define ls (k*2)
  5 #define rs (k*2+1)
  6 #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，又调了半天。。。
  7 using namespace std;
  8 const int MAXN=100099,MAXM=200099;
  9 int n,m,root,mod,num,DFN,op,x,y,z;
 10 int head[MAXN],from[MAXM],to[MAXM],Next[MAXM];
 11 int top[MAXN],w1[MAXN],w2[MAXN],fa[MAXN],son[MAXN],Val[MAXN],dep[MAXN],sz[MAXN],lz[MAXN*5],sum[MAXN*5];
 12 void add(int f,int t)
 13 {
 14   Next[++num]=head[f];
 15   to[num]=t;
 16   head[f]=num;
 17 }
 18 void dfs1(int u,int FA)
 19 {
 20   sz[u]=1;
 21   for(int i=head[u],v=to[i];i;i=Next[i],v=to[i])
 22     if(v!=FA)
 23       {
 24         dep[v]=dep[u]+1;
 25         fa[v]=u;
 26         dfs1(v,u);
 27         sz[u]+=sz[v];
 28         if(sz[v]>sz[son[u]]) son[u]=v;
 29       }
 30 }
 31 void dfs2(int u)
 32 {
 33   w1[u]=++DFN;
 34   if(son[u]!=0)
 35     {
 36       top[son[u]]=top[u];
 37       dfs2(son[u]);
 38     }
 39   for(int i=head[u],v=to[i];i;i=Next[i],v=to[i])
 40     if(v!=son[u] && !w1[v])
 41       {
 42         top[v]=v;
 43         dfs2(v);
 44       }
 45   w2[u]=DFN;
 46 }
 47 void Update(int k,int ll,int rr,int L,int R,int V)
 48 {
 49   if(ll==L && rr==R){sum[k]+=(R-L+1)*V%mod; sum[k]%=mod; lz[k]+=V; return;}
 50   int mid=(ll+rr)/2; down();
 51   if(R<=mid) Update(ls,ll,mid,L,R,V);
 52   else if(mid<L) Update(rs,mid+1,rr,L,R,V);
 53   else Update(ls,ll,mid,L,mid,V) , Update(rs,mid+1,rr,mid+1,R,V);
 54   sum[k]=(sum[ls]+sum[rs])%mod;
 55 }
 56 int Query(int k,int ll,int rr,int L,int R)
 57 {
 58   if(ll==L && rr==R){return sum[k];}
 59   int mid=(ll+rr)/2; down();
 60   if(R<=mid) return Query(ls,ll,mid,L,R);
 61   else if(mid<L) return Query(rs,mid+1,rr,L,R);
 62   else return (Query(ls,ll,mid,L,mid)+Query(rs,mid+1,rr,mid+1,R))%mod;
 63 }
 64 int main()
 65 {
 66   scanf("%d%d%d%d",&n,&m,&root,&mod);
 67   for(int i=1;i<=n;i++) scanf("%d",&Val[i]);
 68   for(int i=1,a,b;i<n;i++)
 69     {
 70       scanf("%d%d",&a,&b);
 71       add(a,b); add(b,a);
 72     }
 73   top[root]=root;
 74   dep[root]=1;
 75   dfs1(root,0);
 76   dfs2(root);
 77   for(int i=1;i<=n;i++) Update(1,1,n,w1[i],w1[i],Val[i]%mod);
 78   while(m--)
 79     {
 80       scanf("%d",&op);
 81       if(op==1)
 82         {
 83           scanf("%d%d%d",&x,&y,&z); z%=mod;
 84           while(top[x]!=top[y])
 85             {
 86               if(dep[top[x]]<dep[top[y]]) swap(x,y);
 87               Update(1,1,n,w1[top[x]],w1[x],z);
 88               x=fa[top[x]];
 89             }
 90           if(dep[x]<dep[y]) swap(x,y);
 91           Update(1,1,n,w1[y],w1[x],z);
 92         }
 93       else if(op==2)
 94         {
 95           scanf("%d%d",&x,&y);
 96           int ans=0;
 97           while(top[x]!=top[y])
 98             {
 99               if(dep[top[x]]<dep[top[y]]) swap(x,y);
100               (ans+=Query(1,1,n,w1[top[x]],w1[x]))%=mod;
101               x=fa[top[x]];
102             }
103           if(dep[x]<dep[y]) swap(x,y);
104           (ans+=Query(1,1,n,w1[y],w1[x]))%=mod;
105           printf("%d\n",ans%mod);
106         }
107       else if(op==3)
108         {
109           scanf("%d%d",&x,&z); z%=mod;
110           Update(1,1,n,w1[x],w2[x],z);
111         }
112       else if(op==4)
113         {
114           scanf("%d",&x);
115           printf("%d\n",Query(1,1,n,w1[x],w2[x]));
116         }
117     }
118   return 0;
119 }

转载于:https://www.cnblogs.com/D-O-Time/p/8040112.html