bzoj 1798: [Ahoi2009]Seq 维护序列seq （线段树）

题目描述

传送门

题目大意：有长为N的数列，不妨设为a1,a2,…,aN 。有如下三种操作形式： (1)把数列中的一段数全部乘一个值; (2)把数列中的一段数全部加一个值; (3)询问数列中的一段数的和，由于答案可能很大，你只需输出这个数模P的值

题解

线段树维护加乘标记的裸题。。。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 400003
#define LL long long 
using namespace std;
int n,m,a[N];
LL tr[N],add[N],mul[N],p;
void update(int x)
{
    tr[x]=(tr[x<<1]+tr[x<<1|1])%p;
}
void build(int now,int l,int r)
{
    mul[now]=1; add[now]=0;
    if (l==r) {
        tr[now]=a[l]; mul[now]=1;
        return;
    }
    int mid=(l+r)/2;
    build(now<<1,l,mid);
    build(now<<1|1,mid+1,r);
    update(now);
}
void pushdown(int now,int l,int r)
{
    int mid=(l+r)/2;
    if (mul[now]!=1) {
        tr[now<<1]=tr[now<<1]*mul[now]%p;
        tr[now<<1|1]=tr[now<<1|1]*mul[now]%p;
        mul[now<<1]=mul[now<<1]*mul[now]%p;
        mul[now<<1|1]=mul[now<<1|1]*mul[now]%p;
        add[now<<1]=add[now<<1]*mul[now]%p;
        add[now<<1|1]=add[now<<1|1]*mul[now]%p;
        mul[now]=1;
    }
    if (add[now]) {
        tr[now<<1]+=add[now]*(mid-l+1)%p;
        if (tr[now]>=p) tr[now]-=p;
        tr[now<<1|1]+=add[now]*(r-mid)%p;
        if (tr[now]>=p) tr[now]-=p;
        add[now<<1]+=add[now];
        if (add[now]>=p) add[now]-=p;
        add[now<<1|1]+=add[now]; 
        if (add[now]>=p) add[now]-=p;
        add[now]=0;
    }
}
void qjmul(int now,int l,int r,int ll,int rr,LL val)
{
    if (ll<=l&&r<=rr) {
        tr[now]=tr[now]*val%p;
        mul[now]=mul[now]*val%p;
        add[now]=add[now]*val%p;
        return;
    }
    int mid=(l+r)/2;
    pushdown(now,l,r);
    if (ll<=mid) qjmul(now<<1,l,mid,ll,rr,val);
    if (rr>mid) qjmul(now<<1|1,mid+1,r,ll,rr,val);
    update(now);
}
void qjadd(int now,int l,int r,int ll,int rr,LL val)
{
    if (ll<=l&&r<=rr) {
        tr[now]=(LL)(tr[now]+(LL)(r-l+1)*val%p);
        if (tr[now]>=p) tr[now]-=p;
        add[now]+=val;
        if (add[now]>=p) add[now]-=p;
        return;
    }
    int mid=(l+r)/2;
    pushdown(now,l,r);
    if (ll<=mid) qjadd(now<<1,l,mid,ll,rr,val);
    if (rr>mid) qjadd(now<<1|1,mid+1,r,ll,rr,val);
    update(now);
    //cout<<l<<" "<<r<<" "<<tr[now]<<endl;
}
LL qjsum(int now,int l,int r,int ll,int rr)
{
    if (ll<=l&&r<=rr) return tr[now];
    int mid=(l+r)/2; LL ans=0;
    pushdown(now,l,r);
    if (ll<=mid) ans+=qjsum(now<<1,l,mid,ll,rr); ans%=p;
    if (rr>mid) ans+=qjsum(now<<1|1,mid+1,r,ll,rr); ans%=p;
    return ans;
}
int main()
{
    freopen("a.in","r",stdin);
    //freopen("my.out","w",stdout);
    scanf("%d%lld",&n,&p);
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    build(1,1,n);
    scanf("%d",&m);
    for (int i=1;i<=m;i++) {
        int opt,x,y,z;
        scanf("%d%d%d",&opt,&x,&y);
        if (opt==2) {
            scanf("%d",&z);
            qjadd(1,1,n,x,y,z);
        //  cout<<tr[1]<<endl;
        }
        if (opt==1) {
            scanf("%d",&z);
            qjmul(1,1,n,x,y,z);
        }
        if (opt==3) {
            printf("%lld\n",qjsum(1,1,n,x,y)%p);
        }
    }
}