luogu_3373 【模板】线段树 2
#include <cstdio> #include <iostream> using namespace std; const int N=1000010; int n,m,mod; long long a[N],addv[N],mulv[N]; void pushdown(int o,int len){ if(mulv[o]!=1){ mulv[o*2]=mulv[o*2]*mulv[o]%mod; mulv[o*2+1]=mulv[o*2+1]*mulv[o]%mod; a[o*2]=a[o*2]*mulv[o]%mod; a[o*2+1]=a[o*2+1]*mulv[o]%mod; addv[o*2]=addv[o*2]*mulv[o]%mod; addv[o*2+1]=addv[o*2+1]*mulv[o]%mod; mulv[o]=1; } if(addv[o]){ addv[o*2]=(addv[o*2]+addv[o])%mod; addv[o*2+1]=(addv[o*2+1]+addv[o])%mod; a[o*2]=(a[o*2]+addv[o]*(len-len/2))%mod; a[o*2+1]=(a[o*2+1]+addv[o]*(len-len/2))%mod; addv[o]=0; } return; } void pushup(int o){ a[o]=(a[o*2]+a[o*2+1])%mod; return; } void update(int l,int r,int L,int R,long long v,int o){ if(l>R||r<L)return; if(L<=l && r<=R){addv[o]=(addv[o]+v)%mod; a[o]=(a[o]+v*(r-l+1))%mod; return;} pushdown(o,r-l+1); int mid=(l+r)>>1; update(l,mid,L,R,v,o*2); update(mid+1,r,L,R,v,o*2+1); pushup(o); } int query(int l,int r,int L,int R,int o){ int ans=0; if(l>R||r<L)return 0; if(L<=l && r<=R)return a[o]; pushdown(o,r-l+1); int mid=(l+r)>>1; ans+=query(l,mid,L,R,o*2)%mod; ans+=query(mid+1,r,L,R,o*2+1)%mod; return ans%mod; } void mul(int l,int r,int L,int R,int v,int o){ if(l>R||r<L)return; if(L<=l && r<=R){mulv[o]=mulv[o]*v%mod; a[o]=(a[o]*v)%mod; addv[o]=addv[o]*v%mod; return;} pushdown(o,r-l+1); int mid=(l+r)>>1; mul(l,mid,L,R,v,o*2); mul(mid+1,r,L,R,v,o*2+1); pushup(o); } int main(){ int k=0; scanf("%d%d%d",&n,&m,&mod); while((1<<k)<n)k++; k=1<<k; for(int i=k;i<k+n;i++)scanf("%lld",&a[i]); for(int i=k-1;i;i--)a[i]=a[i*2]+a[i*2+1]; for(int i=1;i<k+n;i++)mulv[i]=1; while(m--){ int x; scanf("%d",&x); if(x==2){ int l,r; long long v; scanf("%d%d%lld",&l,&r,&v); v%=mod; update(1,k,l,r,v,1); } else if(x==3){ int l,r; scanf("%d%d",&l,&r); printf("%d\n",query(1,k,l,r,1)); } else { int l,r; long long v; scanf("%d%d%lld",&l,&r,&v); v%=mod; mul(1,k,l,r,v,1); } } return 0; }