题意:对一串序列有如下操作。
1.将某个区间的每个数都加上某个数
2.将某个区间的每个数乘上某个数
3.询问区间和
做法:线段树,维护时记录三个数组,一起下传。
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
typedef long long LL;
const int maxn=100010;
LL mod;
int n,p,sak[maxn];
LL sum[maxn<<2];
LL add[maxn<<2],mul[maxn<<2];
int ql,qr;
void push_up(int o){
sum[o]=(sum[o<<1]+sum[o<<1|1])%mod;
}
void create_tree(int o,int l,int r){
mul[o]=1;sum[o]=0;
if(l==r){
scanf("%lld",&sum[o]);
sum[o]=sum[o]%mod;
return;
}
int mid=(l+r)>>1;
create_tree(o<<1,l,mid);
create_tree(o<<1|1,mid+1,r);
push_up(o);
}
void Mod(int o){
sum[o]%=mod;
add[o]%=mod;
mul[o]%=mod;
}
void push_down(int o,int l,int r,int mid){
int u=o<<1,v=o<<1|1;
add[u]=add[u]*mul[o]+add[o];
add[v]=add[v]*mul[o]+add[o];
mul[u]=mul[o]*mul[u];
mul[v]=mul[o]*mul[v];
sum[u]=sum[u]*mul[o]+add[o]*(mid-l+1);
sum[v]=sum[v]*mul[o]+add[o]*(r-mid);
mul[o]=1;add[o]=0;
Mod(u);Mod(v);
}
void update_mul(int o,int l,int r,int t){
if(ql<=l && qr>=r){
mul[o]*=t;
sum[o]*=t;
add[o]*=t;
Mod(o);
return;
}
int mid=(l+r)>>1;
push_down(o,l,r,mid);
Mod(o<<1),Mod(o<<1|1);
if(ql<=mid)update_mul(o<<1,l,mid,t);
if(qr>mid)update_mul(o<<1|1,mid+1,r,t);
push_up(o);
}
void update_add(int o,int l,int r,int t){
if(ql<=l && qr>=r){
sum[o]+=t*(r-l+1);
add[o]+=t;
Mod(o);
return;
}
int mid=(l+r)>>1;
push_down(o,l,r,mid);
Mod(o<<1);Mod(o<<1|1);
if(ql<=mid)update_add(o<<1,l,mid,t);
if(qr>mid)update_add(o<<1|1,mid+1,r,t);
push_up(o);
}
int query(int o,int l,int r){
if(ql<=l && qr>=r){
return sum[o];
}
int mid=(l+r)>>1;
push_down(o,l,r,mid);
Mod(o<<1);Mod(o<<1|1);
int ret=0;
if(ql<=mid) ret+=query(o<<1,l,mid);
if(qr>mid) ret+=query(o<<1|1,mid+1,r);
push_up(o);
return ret%mod;
}
int main(){
scanf("%d%lld",&n,&mod);
create_tree(1,1,n);
int q;
scanf("%d",&q);
while(q--){
int op,t;
scanf("%d%d%d",&op,&ql,&qr);
if(op==1){
scanf("%d",&t);
update_mul(1,1,n,t);
}else if(op==2){
scanf("%d",&t);
update_add(1,1,n,t);
}else{
printf("%d\n",query(1,1,n));
}
}
return 0;
}
^_^