bzoj1798: [Ahoi2009]Seq 维护序列seq

bzoj1798: [Ahoi2009]Seq 维护序列seq
又一道裸线段树。
交上去一直RERERE，调了半天发现把叶子节点pushdown了可不RE。真是代码力低下。
直接上AC代码。

---

代码

#include<iostream>
#include<cstdio>
#define mxn 100016
using namespace std;
int n,m,a[mxn],x,y,p,op,val;
int ll[mxn<<2],rr[mxn<<2];
long long ad[mxn<<2],mu[mxn<<2],sum[mxn<<2];
void update(int k)
{sum[k]=(sum[k<<1]+sum[k<<1|1])%p;}
void build(int k,int l,int r)
{
    ll[k]=l,rr[k]=r,ad[k]=0,mu[k]=1;
    if(l==r)sum[k]=a[l]%p;
    else
    {
        int mid=(l+r)>>1;
        build(k<<1,l,mid),build(k<<1|1,mid+1,r);
        update(k);
    }
}
void pushdown(int k)
{
    mu[k<<1]=(mu[k<<1]*mu[k])%p,mu[k<<1|1]=(mu[k<<1|1]*mu[k])%p;
    ad[k<<1]=(ad[k]+ad[k<<1]*mu[k])%p,ad[k<<1|1]=(ad[k]+ad[k<<1|1]*mu[k])%p;
    sum[k<<1]=(sum[k<<1]*mu[k]%p+ad[k]*(rr[k<<1]-ll[k<<1]+1))%p;
    sum[k<<1|1]=(sum[k<<1|1]*mu[k]%p+ad[k]*(rr[k<<1|1]-ll[k<<1|1]+1))%p;
    ad[k]=0,mu[k]=1;
}
void add(int k,int l,int r,int v)
{
    sum[k]=(sum[k]+v*(r-l+1))%p;
    if(ll[k]==l&&rr[k]==r)
    ad[k]=(ad[k]+v)%p;
    else
    {
        pushdown(k);
        int mid=(ll[k]+rr[k])>>1;
        if(l>mid)add(k<<1|1,l,r,v);
        else if(r<=mid)add(k<<1,l,r,v);
        else add(k<<1,l,mid,v),add(k<<1|1,mid+1,r,v);
    }
}
void mul(int k,int l,int r,int v)
{
    if(ll[k]!=rr[k])
    pushdown(k);
    if(ll[k]==l&&rr[k]==r)
    sum[k]=sum[k]*v%p,mu[k]=mu[k]*v%p;
    else
    {
        int mid=(ll[k]+rr[k])>>1;
        if(l>mid)mul(k<<1|1,l,r,v);
        else if(r<=mid)mul(k<<1,l,r,v);
        else mul(k<<1,l,mid,v),mul(k<<1|1,mid+1,r,v);
        update(k);
    }
}
int qry(int k,int l,int r)
{
    if(ll[k]==l&&rr[k]==r)return sum[k];
    pushdown(k);
    int mid=(ll[k]+rr[k])>>1;
    if(l>mid)return qry(k<<1|1,l,r);
    else if(r<=mid)return qry(k<<1,l,r);
    else return(qry(k<<1,l,mid)+qry(k<<1|1,mid+1,r))%p;
}
int main()
{
    scanf("%d%d",&n,&p);
    for(int i=1;i<=n;i++)
    scanf("%d",&a[i]);
    build(1,1,n);
    scanf("%d",&m);
    while(m--)
    {
        scanf("%d%d%d",&op,&x,&y);
        if(op==1)
        scanf("%d",&val),mul(1,x,y,val%p);
        else if(op==2)
        scanf("%d",&val),add(1,x,y,val%p);
        else printf("%d\n",qry(1,x,y));
    }
}