洛谷P2023 [AHOI2009] 维护序列 线段树

给出$n\le 1e5$个数，然后要求支持区间加，区间乘，区间求和。答案对$P$取模。
比最简单的线段树多了一个区间乘，用两个$lazy$标记分别维护加法和乘法。对于加法，暴力修改当前线段和传递标记与之前是一致的。而对于乘法，不但要传递乘法的标记，也要修改加法的标记。这是因为$c(a+b)=ca+cb$。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int N=1e5+7;
int n,q,mod;
ll sum[N<<2],add[N<<2],mul[N<<2];
ll a[N]; 
void pushup(int rt) {
	sum[rt]=sum[rt<<1]+sum[rt<<1|1];
	sum[rt]%=mod;
}
void pushdown(int rt,int l,int mid,int r) {
	if(mul[rt]!=1) {
		sum[rt<<1]=(sum[rt<<1]*mul[rt])%mod;
		sum[rt<<1|1]=(sum[rt<<1|1]*mul[rt])%mod;
		mul[rt<<1]=(mul[rt<<1]*mul[rt])%mod;
		mul[rt<<1|1]=(mul[rt<<1|1]*mul[rt])%mod;
		add[rt<<1]=(add[rt<<1]*mul[rt])%mod;
		add[rt<<1|1]=(add[rt<<1|1]*mul[rt])%mod;
		mul[rt]=1;
	}
	if(add[rt]) {
		sum[rt<<1]=(sum[rt<<1]+add[rt]*(mid-l+1)%mod)%mod;
		sum[rt<<1|1]=(sum[rt<<1|1]+add[rt]*(r-mid)%mod)%mod;
		add[rt<<1]=(add[rt<<1]+add[rt])%mod;
		add[rt<<1|1]=(add[rt<<1|1]+add[rt])%mod;
		add[rt]=0;
	}
}
void build(int rt,int l,int r) {
	mul[rt]=1,add[rt]=0; 
	if(l==r) { sum[rt]=a[l],mul[rt]=1,add[rt]=0; return; }
	int mid=(l+r)>>1;
	build(rt<<1,l,mid);
	build(rt<<1|1,mid+1,r);
	pushup(rt);
}
void ADD(int rt,int l,int r,int L,int R,int x) {
	if(R<l||L>r) return;
	if(L<=l&&r<=R) {
		sum[rt]=(sum[rt]+1LL*x*(r-l+1)%mod)%mod;
		add[rt]=(add[rt]+x)%mod;
		return;
	}
	int mid=(l+r)>>1;
	pushdown(rt,l,mid,r);
	ADD(rt<<1,l,mid,L,R,x);
	ADD(rt<<1|1,mid+1,r,L,R,x);
	pushup(rt);
}
void MUL(int rt,int l,int r,int L,int R,int x) {
	if(R<l||L>r) return;
	if(L<=l&&r<=R) {
		sum[rt]=(sum[rt]*x)%mod;
		mul[rt]=(mul[rt]*x)%mod;
		add[rt]=(add[rt]*x)%mod;
		return;
	}
	int mid=(l+r)>>1;
	pushdown(rt,l,mid,r);
	MUL(rt<<1,l,mid,L,R,x);
	MUL(rt<<1|1,mid+1,r,L,R,x);
	pushup(rt);
}
ll query(int rt,int l,int r,int L,int R) {
	if(R<l||L>r) return 0;
	if(L<=l&&r<=R) return sum[rt];
	int mid=(l+r)>>1;
	pushdown(rt,l,mid,r);
	return (query(rt<<1,l,mid,L,R)+query(rt<<1|1,mid+1,r,L,R))%mod;
}
int main() {
	scanf("%d%d",&n,&mod);
	for(int i=1;i<=n;i++)
		scanf("%lld",&a[i]);
	build(1,1,n);
	scanf("%d",&q); 
	while(q--) {
		int opt,l,r,x; 
		scanf("%d",&opt);
		if(opt==1) {
			scanf("%d%d%d",&l,&r,&x);
			MUL(1,1,n,l,r,x);
		}
		else if(opt==2) {
			scanf("%d%d%d",&l,&r,&x);
			ADD(1,1,n,l,r,x);
		}
		else if(opt==3) {
			scanf("%d%d",&l,&r);
			printf("%lld\n",query(1,1,n,l,r));
		}
	} 
	return 0;
}