luoguP3373 线段树模板2(懒惰标记的优先级问题)

在这里插入图片描述
在这里插入图片描述
这个题在之前的模板1的基础上多加了一种懒惰标记,而多加的这一种懒惰标记引起了优先级的问题,到底是应该先加后乘呢还是应该先乘后加呢?按理说两种都能实现,但是先乘后加的方法更利于更新,所以这里写了先乘后加的方法,需要再新开一个数组来储存所有的乘法操作,由于是先乘后加,所以每次乘法操作的数组作出更新时,加法操作的数组也应该进行相应的更新以确保懒惰标记的正确性(有多种懒惰标记时一定要注意它们之间的关联关系),还有,记得对所有可能溢出的地方进行取模。
下面是代码:

#include<iostream>
using namespace std;
typedef long long ll;
int p;
const int maxn = 100005;
ll dat[maxn];//储存数据 
ll tree[maxn<<2];//储存线段树的数组常开成数据的4倍大小
ll add[maxn<<2];//用来标记区间加减的操作 
ll multy[maxn<<2];//用来标记区间乘法操作
void initmulty()
{
	for(int i=0;i<(maxn<<2);++i)
		multy[i]=1;//乘法操作,数组初始化为1 
}
void pushup(int node)//用于更新线段树的每个结点,在这里的含义是区间和的合并 
{
	tree[node]=(tree[node<<1]+tree[node<<1|1])%p;//记得%p 
}
void build(int l,int r,int node)//建树 
{
	if(l==r)//递归出口时到达叶节点时 
	{
		tree[node]=dat[l];
		return; 
	}
	int mid=(l+r)>>1;
	build(l,mid,node<<1);//递归建立左子树 
	build(mid+1,r,node<<1|1);//递归建立右子树
	pushup(node);//更新结点信息 
}
void point_update(int l,int r,int node,int dat_i,int c)//这是点修改函数,dat_i是dat数组的下标,作出某种改变用到的数值 
{
	if(l==r)//到达叶节点时作出某些修改,这里是+=操作 
	{
		tree[node]+=c;
		return; 
	}
	int mid=(l+r)>>1;
	if(dat_i<=mid) point_update(l,mid,node<<1,dat_i,c);//根据条件判断应该往左子树调用还是往右子树调用 
	else point_update(mid+1,r,node<<1|1,dat_i,c);
	pushup(node);//子节点更新,路径上的节点都应该更新 
}
void pushdown(int node,int ln,int rn)//这是下推懒惰标记(即延时标记,是一种暂时不处理,等到用到的时候再进行处理的思想)函数,ln,rn是左子树、右子树对应的线段长度,这个函数的作用是下推标记实现区间修改 
{
	//下推标记到左右子树,其目的是间接实现标记叠加,即不影响下一个标记 
	multy[node<<1]=(multy[node<<1]*multy[node])%p;//先乘后加,注意每次multy数组的改变都可能会引起add数组的改变,这些操作实际上是在维护懒惰标记的正确性 
	multy[node<<1|1]=(multy[node<<1|1]*multy[node])%p;
	add[node<<1]=(add[node<<1]*multy[node]+add[node])%p;
	add[node<<1|1]=(add[node<<1|1]*multy[node]+add[node])%p;
	//更新子节点 
	tree[node<<1]=(tree[node<<1]*multy[node]+add[node]*ln)%p;
	tree[node<<1|1]=(tree[node<<1|1]*multy[node]+add[node]*rn)%p;
	//清除本节点的标记,为下个标记作准备 
	add[node]=0; 
	multy[node]=1;
}
void interval_update(int l,int r,int node,int start,int end,int c)//这是区间修改函数,更新从start到end的值 
{
	if(start<=l&&end>=r)//如果当前区间完全在操作区间[start,end]内 
	{
		tree[node]=(tree[node]+c*(r-l+1))%p;//更新当前节点
		add[node]=(add[node]+c)%p;//作标记以对下面的节点进行修改
		return; 
	}
	int mid = (l+r)>>1;
	pushdown(node,mid-l+1,r-mid);//下推标记,即对下面的节点进行修改
	//根据条件判断往哪个方向进行递归 
	if(start<=mid) interval_update(l,mid,node<<1,start,end,c);
	if(end>mid) interval_update(mid+1,r,node<<1|1,start,end,c);
	pushup(node);//更新节点 
}
void interval_update_m(int l,int r,int node,int start,int end,int c)//乘法的区间修改函数 
{
	if(start<=l&&end>=r)
	{
		tree[node]=(c*tree[node])%p; 
		multy[node]=(multy[node]*c)%p;//multy改变,add也应该作出相应改变 
		add[node]=(add[node]*c)%p;
		return;
	}
	int mid=(l+r)>>1;
	pushdown(node,mid-l+1,r-mid);//下推标记 
	if(start<=mid) interval_update_m(l,mid,node<<1,start,end,c);
	if(end>mid) interval_update_m(mid+1,r,node<<1|1,start,end,c);
	pushup(node);//更新节点 
}
ll query(int l,int r,int node,int start,int end)//区间询问函数 
{
	if(start<=l&&end>=r)//在区间内,直接返回
	{
		return tree[node]%p;
	}
	int mid=(l+r)>>1;
	pushdown(node,mid-l+1,r-mid);//下推标记确保sum正确
	//统计答案
	ll ans=0;
	//根据条件选择递归方向 
	if(start<=mid) ans=(ans%p+query(l,mid,node<<1,start,end)%p)%p;
	if(end>mid) ans=(ans%p+query(mid+1,r,node<<1|1,start,end)%p)%p;
	return ans%p;
}
int main()
{
	initmulty();
	int n,m,q,x,y,k;
	cin>>n>>m>>p;
	for(int i=1;i<=n;++i)
	{
		cin>>dat[i]; 
	}
	build(1,n,1);//建树 
	for(int i=1;i<=m;++i)
	{
		cin>>q;
		if(q==2)
		{
			cin>>x>>y>>k;
			interval_update(1,n,1,x,y,k);
		}
		else if(q==3)
		{
			cin>>x>>y;
			cout<<query(1,n,1,x,y)<<endl;
		}
		else
		{
			cin>>x>>y>>k;
			interval_update_m(1,n,1,x,y,k);
		}
	}
	return 0;
}
/*
5 5 38
1 5 4 2 3
2 1 4 1
3 2 5
1 2 4 2
2 3 5 5
3 1 4

17
2
*/
/*
8 10 571373
5929 7152 8443 6028 8580 5449 8473 4237 
2 4 8 4376
1 2 8 9637
2 2 6 7918
2 5 8 5681
3 2 8
1 1 5 6482
3 1 5
1 5 8 8701
2 5 8 7992
2 5 8 7806

478836
562114
*/
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值