这个题在之前的模板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
*/