【分块】bzoj1798 [Ahoi2009]Seq 维护序列seq

分块，打标记，维护两个标记：乘的 和 加的。

每次 区间乘的时候，对 乘标记 和 加标记 都 乘上那个值。

每次 区间加的时候 对 加标记 加上那个值。

(ax+b)*v=axv+bv。开 long long。

#include<cstdio>
#include<cmath>
using namespace std;
typedef long long ll;
int n,sum,sz,num[100001],l[350],r[350],x,y,m,op;
ll MOD,a[100001],sumv[350],lzy1[350],lzy2[350],v;
void makeblock()
{
    sz=sqrt(n); if(!sz) sz=1;
    for(sum=1;sum*sz<n;sum++)
      {
          l[sum]=r[sum-1]+1; r[sum]=sum*sz;
          for(int i=l[sum];i<=r[sum];++i)
          {
              num[i]=sum;
              sumv[sum]=(sumv[sum]+a[i])%MOD;
          }
      }
    l[sum]=r[sum-1]+1; r[sum]=n;
    for(int i=l[sum];i<=r[sum];++i)
      {
          num[i]=sum;
          sumv[sum]=(sumv[sum]+a[i])%MOD;
      }
    for(int i=1;i<=sum;++i) lzy1[i]=1;
}
void pushdown(const int &p)
{
    if(lzy1[p]!=1 || lzy2[p])
      {
          for(int i=l[p];i<=r[p];++i)
          a[i]=(a[i]*lzy1[p]+lzy2[p])%MOD;
          lzy1[p]=1; lzy2[p]=0;
      }
}
void work1(const int &L,const int &R)
{
    sumv[num[L]]=0;
    for(int i=L;i<=R;++i) a[i]=(a[i]*v)%MOD;
    for(int i=l[num[L]];i<=r[num[L]];++i)
      sumv[num[L]]=(sumv[num[L]]+a[i])%MOD;
}
void update1()
{
    pushdown(num[x]); pushdown(num[y]);
    if(num[x]==num[y]) work1(x,y);
    else
      {
          work1(x,r[num[x]]); work1(l[num[y]],y);
          for(int i=num[x]+1;i<num[y];++i)
            {
                lzy1[i]=(lzy1[i]*v)%MOD;
                lzy2[i]=(lzy2[i]*v)%MOD;
                sumv[i]=(sumv[i]*v)%MOD;
            }
      }
}
void work2(const int &L,const int &R)
{
    for(int i=L;i<=R;++i) a[i]=(a[i]+v)%MOD;
    sumv[num[L]]=(sumv[num[L]]+(ll)(R-L+1)*v)%MOD;
}
void update2()
{
    pushdown(num[x]); pushdown(num[y]);
    if(num[x]==num[y]) work2(x,y);
    else
      {
          work2(x,r[num[x]]); work2(l[num[y]],y);
          for(int i=num[x]+1;i<num[y];++i)
          {
              lzy2[i]=(lzy2[i]+v)%MOD;
              sumv[i]=(sumv[i]+(ll)sz*v)%MOD;
          }
      }
}
void query()
{
    pushdown(num[x]); pushdown(num[y]); ll ans=0;
    if(num[x]==num[y]) {for(int i=x;i<=y;++i) ans=(ans+a[i])%MOD;}
    else
      {
          for(int i=x;i<=r[num[x]];++i) ans=(ans+a[i])%MOD;
          for(int i=l[num[y]];i<=y;++i) ans=(ans+a[i])%MOD;
          for(int i=num[x]+1;i<num[y];++i) ans=(ans+sumv[i])%MOD;
      } printf("%d
",(int)ans);
}
int main()
{
    scanf("%d%lld",&n,&MOD);
    for(int i=1;i<=n;++i) scanf("%lld",&a[i]);
    makeblock(); scanf("%d",&m);
    for(;m>0;--m)
      {
          scanf("%d%d%d",&op,&x,&y);
          if(op==1) {scanf("%lld",&v); update1();}
          else if(op==2) {scanf("%lld",&v); update2();}
          else query();
      }
    return 0;
}