bzoj 1798 [Ahoi2009]Seq 维护序列seq ——线段树

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=1798

先乘后加，就可给加法标记乘上乘法标记。

注意可能有 *0 的操作，所以 pshd 时不是 cg[ cr ]>1 而是 cg[ cr ]!=1 。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
#define ls Ls[cr]
#define rs Rs[cr]
using namespace std;
const int N=1e5+,M=N<<;
int n,m,mod,a[N],Ls[M],Rs[M],cg[M],jg[M],sm[M],tot;
int rdn()
{
  int ret=;bool fx=;char ch=getchar();
  while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
  while(ch>=''&&ch<='') ret=(ret<<)+(ret<<)+ch-'',ch=getchar();
  return fx?ret:-ret;
}
int g[];
void wrt(int x)
{
  if(x<)putchar('-'),x=-x;
  if(!x){puts("");return;}
  int t=;while(x)g[++t]=x%,x/=;
  while(t)putchar(g[t]+''),t--;puts("");
}
void upd(int &x){x>=mod?x-=mod:;}
void pshp(int cr){sm[cr]=sm[ls]+sm[rs];upd(sm[cr]);}
void build(int l,int r,int cr)
{
  cg[cr]=;
  if(l==r){sm[cr]=a[l]%mod;return;}
  int mid=l+r>>;
  ls=++tot; build(l,mid,ls);
  rs=++tot; build(mid+,r,rs);
  pshp(cr);
}
void pshd(int cr,int l,int mid,int r)
{
  if(cg[cr]!=)
    {
      int w=cg[cr]; cg[cr]=;
      cg[ls]=(ll)cg[ls]*w%mod; cg[rs]=(ll)cg[rs]*w%mod;
      jg[ls]=(ll)jg[ls]*w%mod; jg[rs]=(ll)jg[rs]*w%mod;
      sm[ls]=(ll)sm[ls]*w%mod; sm[rs]=(ll)sm[rs]*w%mod;
    }
  if(jg[cr])
    {
      int w=jg[cr]; jg[cr]=;
      jg[ls]+=w;upd(jg[ls]); jg[rs]+=w;upd(jg[rs]);
      sm[ls]=(sm[ls]+(ll)(mid-l+)*w)%mod; sm[rs]=(sm[rs]+(ll)(r-mid)*w)%mod;
    }
}
void mdfy(int l,int r,int cr,int L,int R,int k,bool fx)
{
  if(l>=L&&r<=R)
    {
      if(!fx)
    cg[cr]=(ll)cg[cr]*k%mod, sm[cr]=(ll)sm[cr]*k%mod, jg[cr]=(ll)jg[cr]*k%mod;
      else
    jg[cr]+=k,upd(jg[cr]),sm[cr]=(sm[cr]+(ll)(r-l+)*k)%mod;
      return;
    }
  int mid=l+r>>; pshd(cr,l,mid,r);
  if(L<=mid)mdfy(l,mid,ls,L,R,k,fx);
  if(mid<R)mdfy(mid+,r,rs,L,R,k,fx);
  pshp(cr);
}
int query(int l,int r,int cr,int L,int R)
{
  if(l>=L&&r<=R)return sm[cr];
  int mid=l+r>>,ret=; pshd(cr,l,mid,r);
  if(L<=mid)ret=query(l,mid,ls,L,R);
  if(mid<R)ret+=query(mid+,r,rs,L,R),upd(ret);
  return ret;
}
int main()
{
  n=rdn(); mod=rdn();
  for(int i=;i<=n;i++)a[i]=rdn();
  tot=; build(,n,);
  m=rdn();
  for(int i=,op,l,r,k;i<=m;i++)
    {
      op=rdn();l=rdn();r=rdn();
      if(op==)
    {
      k=rdn()%mod;mdfy(,n,,l,r,k,);
    }
      if(op==)
    {
      k=rdn()%mod;mdfy(,n,,l,r,k,);
    }
      if(op==)
    wrt(query(,n,,l,r));
    }
  return ;
}