P3384(树链剖分，入门，模板）

题目描述

题目详见：https://www.luogu.org/problemnew/show/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为根节点的子树内所有节点值之和

解题思路：第一次打树链剖分，其实挺简单的，详见：https://www.luogu.org/problemnew/solution/P3384讲的很清楚

 

 

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define N 100009
using namespace std;
struct node
{
	int v,next;
}s[2*N];
struct tree
{
	int l,r,w,c;
}p[3*N];
int w[N],head[N],cnt,fa[N],siz[N],son[N],dep[N],top[N],id[N],wt[N],mod;
void add_edge(int u,int v)
{
	s[cnt].v=v;
	s[cnt].next=head[u];
	head[u]=cnt++;
	s[cnt].v=u;
	s[cnt].next=head[v];
	head[v]=cnt++;	
}
void dfs1(int u,int f,int deep)
{
	dep[u]=deep;//标记每个点的深度 
    fa[u]=f;//标记每个点的父亲 
    siz[u]=1;//标记每个非叶子节点的子树大小	
    int maxson=-1;
    for(int i=head[u];i!=-1;i=s[i].next)
    {
    	int v=s[i].v;
    	if(v==f) continue;
    	dfs1(v,u,deep+1);
    	siz[u]+=siz[v];
    	if(siz[v]>maxson)
    	{
    		son[u]=v;
    		maxson=siz[v];
		}
	}
}
void dfs2(int u,int t)
{
	id[u]=++cnt;
	wt[cnt]=w[u];
	top[u]=t;
	if(son[u]==-1) return;
	dfs2(son[u],t);
	for(int i=head[u];i!=-1;i=s[i].next)
	{
		int v=s[i].v;
		if(v==fa[u]||v==son[u]) continue;
		dfs2(v,v);
	}
}
void build(int rt,int l,int r)
{
	p[rt].l=l,p[rt].r=r,p[rt].c=0;
	if(l==r)
	{
		p[rt].w=wt[r]%mod;
		return;
	}
	int mid=(l+r)/2;
	build(2*rt,l,mid);
	build(2*rt+1,mid+1,r);
	p[rt].w=(p[2*rt].w+p[2*rt+1].w)%mod;
}
int query(int rt,int l,int r)
{
	if(l==p[rt].l&&r==p[rt].r)
	{
		return p[rt].w%mod;
	}
	if(p[rt].c!=0)
	{
		p[2*rt].c+=p[rt].c;
		p[2*rt+1].c+=p[rt].c;
		p[2*rt].w=(p[2*rt].w+p[rt].c*(p[2*rt].r-p[2*rt].l+1))%mod;
		p[2*rt+1].w=(p[2*rt+1].w+p[rt].c*(p[2*rt+1].r-p[2*rt+1].l+1))%mod;
		p[rt].c=0;
	}
	int mid=(p[rt].l+p[rt].r)/2;
	if(r<=mid)
	{
		return query(2*rt,l,r);
	}else
	if(l>=mid+1)
	{
		return query(2*rt+1,l,r);
	}else
	{
		return (query(2*rt+1,mid+1,r)+query(2*rt,l,mid))%mod;
	}
}
void update(int rt,int l,int r,int kk)
{
	if(l==p[rt].l&&r==p[rt].r)
	{
		p[rt].w+=kk*(r-l+1);
		p[rt].w%=mod;
		p[rt].c+=kk;
		return;
	}
	if(p[rt].c!=0)
	{
		p[2*rt].c+=p[rt].c;
		p[2*rt+1].c+=p[rt].c;
		p[2*rt].w=(p[2*rt].w+p[rt].c*(p[2*rt].r-p[2*rt].l+1))%mod;
		p[2*rt+1].w=(p[2*rt+1].w+p[rt].c*(p[2*rt+1].r-p[2*rt+1].l+1))%mod;
		p[rt].c=0;
	}
	int mid=(p[rt].l+p[rt].r)/2;
	if(r<=mid)
	{
		update(2*rt,l,r,kk);
	}else
	if(l>=mid+1)
	{
		update(2*rt+1,l,r,kk);
	}else
	{
		update(2*rt+1,mid+1,r,kk);
		update(2*rt,l,mid,kk);
	}	
	p[rt].w=(p[2*rt].w+p[2*rt+1].w)%mod;
}
int qrange(int u,int v)
{
	int res=0;
	while(top[u]!=top[v])
	{
		if(dep[top[u]]<dep[top[v]]) swap(u,v);
		res+=query(1,id[top[u]],id[u]);
		res%=mod;
		u=fa[top[u]];
	}
	if(dep[u]>dep[v]) swap(u,v);
	res+=query(1,id[u],id[v]);
	res%=mod;
	return res;
}
void uprange(int u,int v,int k)
{
	k%=mod;
	while(top[u]!=top[v])
	{
		if(dep[top[u]]<dep[top[v]]) swap(u,v);
		update(1,id[top[u]],id[u],k);
		u=fa[top[u]];
	}
	if(dep[u]>dep[v]) swap(u,v);
	update(1,id[u],id[v],k);
}
void addson(int u,int k)
{
	update(1,id[u],id[u]+siz[u]-1,k);
}
int qson(int u)
{
    return query(1,id[u],id[u]+siz[u]-1);
}
int main()
{
	//freopen("t.txt","r",stdin);
	int n,m,r,x,y,a,b,c,tp;
	scanf("%d%d%d%d",&n,&m,&r,&mod);
	for(int i=1;i<=n;i++) scanf("%d",&w[i]);
	cnt=0;
	memset(fa,-1,sizeof(fa));
	memset(head,-1,sizeof(head));
	memset(son,-1,sizeof(son));
	for(int i=0;i<n-1;i++)
	{
		scanf("%d%d",&x,&y);
		add_edge(x,y); 
	}
	dfs1(r,-1,1);
	cnt=0;
	dfs2(r,r);
	build(1,1,n);
	while(m--)
	{
		scanf("%d",&tp);
		if(tp==1)
		{
			scanf("%d%d%d",&a,&b,&c);
			uprange(a,b,c);
		}else
		if(tp==2)
		{
			scanf("%d%d",&a,&b);
			printf("%d\n",qrange(a,b));
		}else
		if(tp==3)
		{
			scanf("%d%d",&a,&b);
			addson(a,b);
		}else
		{
			scanf("%d",&a);
			printf("%d\n",qson(a));
		}
	}
	return 0;
}