树链剖分(模板)

/*冗长的代码还请耐心阅读(两个小时的成果)*/
#include<bits/stdc++.h> using namespace std; #define ls t<<1 #define rs t<<1|1 const int N = 110000; int nxt[N],to[N],id[N],top[N],son[N],sum[N*2],a[N]; int size[N],fa[N],w[N],lazy[N*2],head[N],d[N]; int n,m,cnt,num,t,mod; struct Edge{ int v,nxt; }e[N*2]; void add(int u,int v) { e[++num].v=v; e[num].nxt=head[u]; head[u]=num; } void push(int t) { sum[t]=(sum[ls]+sum[rs])%mod; } void build(int t,int l,int r) { if(l==r) { sum[t]=a[l]; return; } int mid=(l+r)/2; build(ls,l,mid); build(rs,mid+1,r); push(t); } void pushdown(int t,int len) { if(lazy[t]) { lazy[ls]+=lazy[t]; lazy[ls]%=mod; lazy[rs]+=lazy[t]; lazy[rs]%=mod; sum[ls]+=(len-(len/2))*lazy[t]; sum[ls]%=mod; sum[rs]+=(len/2)*lazy[t]; sum[rs]%=mod; lazy[t]=0; } } void modify(int t,int l,int r,int L,int R,int val) { if(L>r || R<l) return; if(L<=l && r<=R) { sum[t]+=val*(r-l+1); sum[t]%=mod; lazy[t]+=val; lazy[t]%=mod; return; } int mid=(l+r)/2; pushdown(t,r-l+1); modify(ls,l,mid,L,R,val); modify(rs,mid+1,r,L,R,val); push(t); } int csum(int t,int l,int r,int L,int R) { if(L>r || R<l) return 0; if(L<=l && r<=R) return sum[t]%mod; int mid=(l+r)/2; pushdown(t,r-l+1); return (csum(ls,l,mid,L,R)%mod+csum(rs,mid+1,r,L,R)%mod)%mod; } void dfs1(int u,int f) { size[u]=1; for(int i=head[u];i;i=e[i].nxt) { int v=e[i].v; if(v!=f) { d[v]=d[u]+1; fa[v]=u; dfs1(v,u); size[u]+=size[v]; if(size[v]>size[son[u]]) son[u]=v; } } } void dfs2(int u,int t) { id[u]=++cnt; a[cnt]=w[u]; top[u]=t; if(son[u]) dfs2(son[u],t); for(int i=head[u];i;i=e[i].nxt) { int v=e[i].v; if(v!=fa[u] && v!=son[u]) dfs2(v,v); } } int calc(int x,int y) { int ans=0; int fx=top[x],fy=top[y]; while(fx!=fy) { if(d[fx]<d[fy]) swap(x,y),swap(fx,fy); ans+=csum(1,1,cnt,id[fx],id[x]); x=fa[fx],fx=top[x]; } if(id[x]>id[y]) swap(x,y); ans+=csum(1,1,cnt,id[x],id[y]); return ans; } void cal(int x,int y,int val) { int fx=top[x],fy=top[y]; while(fx!=fy) { if(d[fx]<d[fy]) swap(x,y),swap(fx,fy); modify(1,1,cnt,id[fx],id[x],val); x=fa[fx],fx=top[x]; } if(id[x]>id[y]) swap(x,y); modify(1,1,cnt,id[x],id[y],val); } int main() { cin>>n>>m>>t>>mod; for(int i=1;i<=n;i++) cin>>w[i],w[i]%=mod; for(int i=1,u,v;i<n;i++) { cin>>u>>v; add(u,v); add(v,u); } d[t]=1; dfs1(t,0);dfs2(t,t);build(1,1,n); for(int i=1,k,x,y,z;i<=m;i++) { cin>>k; if(k==1) { cin>>x>>y>>z; cal(x,y,z%mod); } if(k==2) { cin>>x>>y; cout<<calc(x,y)%mod<<'\n'; } if(k==3) { cin>>x>>y; modify(1,1,cnt,id[x],id[x]+size[x]-1,y%mod); } if(k==4) { cin>>x; cout<<csum(1,1,cnt,id[x],id[x]+size[x]-1)%mod<<endl; } } return 0; }

 

posted @ 2018-12-26 20:20  xmex  阅读(176)  评论(2编辑  收藏  举报