【模板】树链剖分

洛谷传送门：点击打开链接

由于是模板就不说什么了。

这篇博客十分好理解：点击打开链接

时隔几周终于AC的我流下了感动的泪水。各种MLE,TLE,RE,WA。

事实告诉我们要充分理解之后再打数据结构的板子，不然就会像我一样的zz。

题面如下：

 

题目描述

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

输入输出格式

输入格式：

 

 

 

第一行包含4个正整数N、M、R、P，分别表示树的结点个数、操作个数、根节点序号和取模数（即所有的输出结果均对此取模）。

接下来一行包含N个非负整数，分别依次表示各个节点上初始的数值。

接下来N-1行每行包含两个整数x、y，表示点x和点y之间连有一条边（保证无环且连通）

接下来M行每行包含若干个正整数，每行表示一个操作，格式如下：

操作1： 1 x y z

操作2： 2 x y

操作3： 3 x z

操作4： 4 x

 

输出格式：

 

 

 

输出包含若干行，分别依次表示每个操作2或操作4所得的结果（对P取模）

代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<cstdlib>
using namespace std;
const int N=2e5+2;
typedef long long ll;
int n,m,root,rt,cnt,tot,num,mod,res=0;
int f[N],top[N],son[N],dep[N],siz[N],tid[N],rnk[N];
int ini[N],head[N<<1];
struct edge{
	int next,to;
}e[N<<1];
int ad[N<<2],sum[N<<2];

inline int read()
{
	char c=getchar();int x=0,t=1;
	while(c<'0' || c>'9') {if(c=='-') t=-1;c=getchar();}
    while(c>='0' && c<='9') {x=x*10+c-'0';c=getchar();}
    return x*t;
}

inline void pushup(int k){
	sum[k]=(sum[k<<1]+sum[k<<1|1])%mod;
}

inline void pushdown(int k,int len){
	sum[k<<1]+=(len-(len>>1))*ad[k];sum[k<<1]%=mod;
	sum[k<<1|1]+=(len>>1)*ad[k];sum[k<<1|1]%=mod;
	ad[k<<1]+=ad[k];
	ad[k<<1|1]+=ad[k];
	ad[k]=0; 
}

inline void link(int u,int v){
	++tot;
	e[tot].to=v;e[tot].next=head[u];head[u]=tot;
}

inline void df(int k,int d,int fa)
{
	f[k]=fa;dep[k]=d;siz[k]=1;
	for(int i=head[k];i;i=e[i].next){
		if(e[i].to!=fa){
			df(e[i].to,d+1,k);
		    siz[k]+=siz[e[i].to];
		  if(son[k]==0 || siz[e[i].to]>siz[son[k]])
		    son[k]=e[i].to;
		}
	}
}

inline void dfs(int k,int t)
{
	top[k]=t;tid[k]=++cnt;
	rnk[cnt]=ini[k];
	if(son[k]==0) return;
	dfs(son[k],t);
	for(int i=head[k];i;i=e[i].next){
		if(e[i].to!=f[k] && e[i].to!=son[k]){
			  dfs(e[i].to,e[i].to);
		}
	}
}

inline void build(int k,int l,int r)
{

	if(l==r){
		sum[k]=rnk[l];
		if(sum[k]>mod) sum[k]%=mod;
		return;
	}	int mid=(l+r)>>1;
	build(k<<1,l,mid);
	build(k<<1|1,mid+1,r);
	pushup(k);
}

inline void getsum(int k,int l,int r,int L,int R)
{
	int len=r-l+1;
	if(l>=L && r<=R){
		res+=sum[k];res%=mod;return;
	}
	if(ad[k]!=0) pushdown(k,len);
	int mid=(l+r)>>1;
    if(L<=mid)getsum(k<<1,l,mid,L,R);
    if(R>mid)getsum(k<<1|1,mid+1,r,L,R);
}

inline void update(int k,int l,int r,int L,int R,ll adds)
{
	int len=r-l+1;
	if(l>=L && r<=R){
		ad[k]+=adds;sum[k]+=len*adds;
		sum[k]%=mod;
		return;
	}
	if(ad[k]!=0) pushdown(k,len);
	int mid=(l+r)>>1;
	if(L<=mid) update(k<<1,l,mid,L,R,adds);
	if(R>mid) update(k<<1|1,mid+1,r,L,R,adds);
	pushup(k);
}
//以下两个转换中的top,f,x,y要注意 re多次
inline void add(int x,int y,int adds)
{
	adds%=mod;
	while(top[x]!=top[y]){
	   if(dep[top[x]]<dep[top[y]]) swap(x,y);
	   update(1,1,n,tid[top[x]],tid[x],adds);
	   x=f[top[x]];
	}
    if(dep[y]>=dep[x]) update(1,1,n,tid[x],tid[y],adds);
    else update(1,1,n,tid[y],tid[x],adds);
}
 
inline int query(int x,int y)
{
    int ans=0; 
	while(top[x]!=top[y]){
	   if(dep[top[x]]<dep[top[y]]) swap(x,y);
	     res=0;
		 getsum(1,1,n,tid[top[x]],tid[x]);
		 ans+=res;
		  ans%=mod;
		  x=f[top[x]];
	}
    if(dep[x]>dep[y])swap(x,y);
    res=0;
    getsum(1,1,n,tid[x],tid[y]);
    ans+=res;
	return ans%mod;
}

int main(){
	n=read(),m=read(),root=read();mod=read();
	for(int i=1;i<=n;i++) ini[i]=read();
	for(int u,v,i=1;i<n;i++){
		u=read();v=read();
		link(u,v);link(v,u);
	}
	df(root,1,0);dfs(root,root);
	build(1,1,n);
	for(int i=1;i<=m;i++){
		int op,x,y,z; 
		op=read();
		if(op==1){
			x=read(),y=read();z=read();
			add(x,y,z);
		}else if(op==2){
			x=read();y=read();
			printf("%d\n",query(x,y));
		}else if(op==3){
			x=read();z=read();
			update(1,1,n,tid[x],tid[x]+siz[x]-1,z);
		}else{
			x=read();res=0;getsum(1,1,n,tid[x],tid[x]+siz[x]-1);
			printf("%d\n",res);
		}
	}
	return 0;
}