【题解】洛谷P3384【模板】树链剖分

43 篇文章 0 订阅
2 篇文章 0 订阅

题目链接
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述


大佬博客的讲解很清晰。

#include<cstdio>
#include<algorithm>
using namespace std;
template<typename tp>inline void read(tp &x)
{
	x=0;int f=0;char ch=getchar();
	while(ch<'0'||ch>'9')f|=ch=='-',ch=getchar();
	while(ch>='0'&&ch<='9')x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
	if(f)x=-x;
}
template<typename tp>inline void write(tp x)
{
	int buf[40],top=0;
	if(x<0)putchar('-'),x=-x;
	do{
		buf[top++]=x%10;x/=10;
	}while(x);
	for(int i=top-1;i+1;i--)putchar(buf[i]+48);
	putchar('\n');
}
const int N=1e5+10;
int n,m,r,p,w[N],hd[N],id[N],sz[N],fa[N],son[N],tot,cnt,dep[N],top[N],res,c[N<<2],lazy[N<<2],wt[N];
//son[]重儿子编号,id[]新编号,fa[]父亲节点,cnt是dfs序,dep[]深度,sz[]子树大小,top[]当前链顶端节点,res查询答案 
struct Edge{
	int v,nx;
}e[N<<1];
inline void add(int u,int v)
{
	e[++tot].v=v;e[tot].nx=hd[u];hd[u]=tot;
}
void dfs1(int u,int f)
{
	dep[u]=dep[f]+1;fa[u]=f;sz[u]=1;
	int maxson=-1;//记录重儿子的儿子数
	for(int i=hd[u];i;i=e[i].nx)
	{
		int v=e[i].v;
		if(v==f)continue;
		dfs1(v,u);sz[u]+=sz[v];
		if(sz[v]>maxson)son[u]=v,maxson=sz[v];//标记每个节点的重儿子编号 
	} 
}
void dfs2(int u,int topf)//topf是当前链的最顶端的节点 
{
	id[u]=++cnt;wt[cnt]=w[u];//把每个点的初始值赋到新编号上
	top[u]=topf;//这个点所在的顶端
	if(!son[u])return;
	dfs2(son[u],topf);//先处理重儿子,再处理轻儿子的顺序递归处理
	for(int i=hd[u];i;i=e[i].nx) 
	{
		int v=e[i].v;
		if(v==fa[u]||v==son[u])continue;
		dfs2(v,v);//对于每个轻儿子都有一条从它自己开始的链 
	}
}
#define lc (o<<1)
#define rc (o<<1|1)
inline void push_up(int o){c[o]=(c[lc]+c[rc])%p;}
void build(int o,int l,int r)
{
	if(l==r)
	{
		c[o]=wt[l]%p;return;
	}
	int mid=(l+r)>>1;
	build(lc,l,mid);build(rc,mid+1,r);
	push_up(o);
}
inline void push_down(int o,int l,int r)
{
	if(!lazy[o])return;
	int mid=(l+r)>>1;
	lazy[lc]=(lazy[lc]+lazy[o])%p;c[lc]=(c[lc]+lazy[o]*(mid-l+1)%p)%p;
	lazy[rc]=(lazy[rc]+lazy[o])%p;c[rc]=(c[rc]+lazy[o]*(r-mid)%p)%p;
	lazy[o]=0;return;
}
void update(int o,int l,int r,int nl,int nr,int v)
{
	if(nl<=l&&r<=nr)
	{
		lazy[o]=(lazy[o]+v)%p;c[o]=(c[o]+v*(r-l+1)%p)%p;return;
	}
	push_down(o,l,r);
	int mid=(l+r)>>1;
	if(nl<=mid)update(lc,l,mid,nl,nr,v);
	if(nr>mid)update(rc,mid+1,r,nl,nr,v);
	push_up(o);
}
void query(int o,int l,int r,int nl,int nr)
{
	if(nl<=l&&r<=nr)
	{
		res=(res+c[o])%p;return;
	}
    push_down(o,l,r);
    int mid=(l+r)>>1;
    if(nl<=mid)query(lc,l,mid,nl,nr);
    if(nr>mid)query(rc,mid+1,r,nl,nr);
}
inline void updR(int x,int y,int z)
{
	z%=p;
	while(top[x]!=top[y])//当两个点不在同一条链上 
	{
		if(dep[top[x]]<dep[top[y]])swap(x,y);//把x点改为所在链顶端深度更深的那个点
		update(1,1,n,id[top[x]],id[x],z);//更新x到x所在链顶端
		x=fa[top[x]];//x跳到x所在链顶端的那个点的上面那个点 
	}
	if(dep[x]>dep[y])swap(x,y);
	update(1,1,n,id[x],id[y],z);//更新两点之间这条链 
}
inline int queR(int x,int y)
{
	int ans=0;
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		res=0;
		query(1,1,n,id[top[x]],id[x]);//ans加上x到x所在链顶端这一段区间的点权和 
	    ans=(ans+res)%p;x=fa[top[x]];//x跳到x所在链顶端的那个点的上面一个点 
	}
	//直到两个点处于一条链上 
	if(dep[x]>dep[y])swap(x,y);
	res=0;
	query(1,1,n,id[x],id[y]);//这时再加上此时两个点的区间和即可
	ans+=res;return ans%p; 
}
inline void updS(int x,int k)
{
	update(1,1,n,id[x],id[x]+sz[x]-1,k);//子树区间右端点为id[x]+sz[x]-1 
}
inline int queS(int x)
{
	res=0;
	query(1,1,n,id[x],id[x]+sz[x]-1);
	return res;
}
int main()
{
	//freopen("in.txt","r",stdin);
	read(n);read(m);read(r);read(p);
    for(int i=1;i<=n;i++)
	    read(w[i]);
    for(int i=1,u,v;i<n;i++)
        read(u),read(v),add(u,v),add(v,u);
    dfs1(r,0);dfs2(r,r);build(1,1,n);
    while(m--)
    {
    	int k,x,y,z;
    	read(k);
    	if(k==1)
    	{
    		read(x);read(y);read(z);updR(x,y,z);
		}
		else if(k==2)
		{
			read(x);read(y);write(queR(x,y));
		}
		else if(k==3)
		{
			read(x);read(y);updS(x,y);
		}
		else
		{
			read(x);
			write(queS(x));
		}
	}
	return 0;
}

总结

树剖模板

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值