【bzoj1798】维护序列

线段树维护两个标记。

*0的操作在实质上没有任何影响。

#include <cstdio>
#include <cctype>

#define rep(i,a,b) for (int i=(a);i<=(b);i++)

#define LL long long

#define lc (x<<1)
#define rc (x<<1|1)
#define siz(x) (tr[x].r-tr[x].l+1)

const int N=131072;
const int S=262144;

int n,p,m,a[N];
struct Tree {
    int l,r;
    int a,b;        //tag(a,b): ax+b
    int sum;
    Tree(int _l=0,int _r=0) {
        l=_l,r=_r;
        a=1,b=0;
        sum=0;
    }
}tr[S];

int rd(void) {
    int x=0,f=1; char c=getchar();
    while (!isdigit(c)) {
        if (c=='-') f=-1;
        c=getchar();
    }
    while (isdigit(c)) {
        x=x*10+c-'0';
        c=getchar();
    }
    return x*f;
}

void Pushup(int x) {
    tr[x].sum=(tr[lc].sum+tr[rc].sum)%p;
}

void Clear(int x) {
    int a=tr[x].a,b=tr[x].b; if (a==1&&!b) return; tr[x].a=1,tr[x].b=0;
    tr[lc].sum=((LL)a*tr[lc].sum+(LL)b*siz(lc))%p; tr[lc].a=(LL)a*tr[lc].a%p; tr[lc].b=((LL)a*tr[lc].b+b)%p;
    tr[rc].sum=((LL)a*tr[rc].sum+(LL)b*siz(rc))%p; tr[rc].a=(LL)a*tr[rc].a%p; tr[rc].b=((LL)a*tr[rc].b+b)%p;
}

void Build(int x,int l,int r) {
    tr[x]=Tree(l,r);
    if (l!=r) {
        int mid=(l+r)>>1;
        Build(lc,l,mid);
        Build(rc,mid+1,r);
        Pushup(x);
    }
    else tr[x].sum=a[l];
}

void Modify(int x,int l,int r,int a,int b) {
    if (l<=tr[x].l&&tr[x].r<=r) {
        tr[x].sum=((LL)a*tr[x].sum+(LL)b*siz(x))%p;
        tr[x].a=(LL)a*tr[x].a%p;
        tr[x].b=((LL)a*tr[x].b+b)%p;
        return;
    }
    Clear(x);
    int mid=(tr[x].l+tr[x].r)>>1;
    if (l<=mid)
        Modify(lc,l,r,a,b);
    if (mid<r)
        Modify(rc,l,r,a,b);
    Pushup(x);
}

int Query(int x,int l,int r) {
    if (l<=tr[x].l&&tr[x].r<=r) return tr[x].sum;
    Clear(x);
    int sum=0; int mid=(tr[x].l+tr[x].r)>>1;
    if (l<=mid) {
        int t=Query(lc,l,r);
        sum=(sum+t)%p;
    }
    if (mid<r) {
        int t=Query(rc,l,r);
        sum=(sum+t)%p;
    }
    return sum;
}

int main(void) {
    #ifndef ONLINE_JUDGE
    freopen("sd.in","r",stdin);
    freopen("sd.out","w",stdout);
    #endif

    n=rd(),p=rd();
    rep(i,1,n) a[i]=rd();
    Build(1,1,n);

    m=rd();
    rep(i,1,m) {
        int kd=rd(); int x,y,z;
        switch (kd) {
            case 1:
                x=rd(),y=rd(),z=rd();
                Modify(1,x,y,z,0);
                break;
            case 2:
                x=rd(),y=rd(),z=rd();
                Modify(1,x,y,1,z);
                break;
            case 3:
                x=rd(),y=rd();
                int res=Query(1,x,y);
                printf("%d\n",res);
                break;
        }
    }

    return 0;
}