BZOJ_1798_[AHOI2009]维护序列_线段树

BZOJ_1798_[AHOI2009]维护序列_线段树

题意：老师交给小可可一个维护数列的任务，现在小可可希望你来帮他完成。 有长为N的数列，不妨设为a1,a2,…,aN 。有如下三种操作形式： (1)把数列中的一段数全部乘一个值; (2)把数列中的一段数全部加一个值; (3)询问数列中的一段数的和，由于答案可能很大，你只需输出这个数模P的值。

分析：线段树上要打两个标记。要注意下传的顺序。显然先乘后加和先加后乘是不一样的。我们发现如果是先加后乘的话更改子树值的式子里会出现除法。不妨规定任何时候都先乘后加。推出的式子即为

t[lson]=(t[lson]*mul[pos]+add[pos]*(mid-l+1))%p;

mul[lson]=(mul[lson]*mul[pos])%p;

add[lson]=(add[lson]*mul[pos]+add[pos])%p;

代码：

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define ls p<<1
#define rs p<<1|1
#define LL long long
#define N 100050
LL add[N<<2],t[N<<2],mul[N<<2],mod;
int a[N],n,m;
void bt(int l,int r,int p){
    mul[p]=1;
    if(l==r){
        scanf("%lld",&t[p]);return ;
    }
    int mid=l+r>>1;
    bt(l,mid,ls);bt(mid+1,r,rs);
    t[p]=(t[ls]+t[rs])%mod;
}
void pud(int l,int r,int p){
    int mid=l+r>>1;
    if(add[p]==0&&mul[p]==1)return ;
    t[ls]=(t[ls]*mul[p]+add[p]*(mid-l+1))%mod;
    t[rs]=(t[rs]*mul[p]+add[p]*(r-mid))%mod;
    mul[ls]=mul[ls]*mul[p]%mod;
    mul[rs]=mul[rs]*mul[p]%mod;
    add[ls]=(add[ls]*mul[p]+add[p])%mod;
    add[rs]=(add[rs]*mul[p]+add[p])%mod;
    mul[p]=1;add[p]=0;
}
void upad(int l,int r,int x,int y,int c,int p){
    if(x<=l&&y>=r){
        add[p]=(add[p]+c)%mod;
        t[p]+=1ll*(r-l+1)*c;t[p]%=mod;
        return ;
    }
    int mid=l+r>>1;
    pud(l,r,p);
    if(x<=mid)upad(l,mid,x,y,c,ls);
    if(y>mid)upad(mid+1,r,x,y,c,rs);
    t[p]=(t[ls]+t[rs])%mod;
}
void upmu(int l,int r,int x,int y,int c,int p){
    if(x<=l&&y>=r){
        mul[p]=mul[p]*c%mod;
        add[p]=add[p]*c%mod;
        t[p]=t[p]*c%mod;
        return ;
    }
    pud(l,r,p);
    int mid=l+r>>1;
    if(x<=mid)upmu(l,mid,x,y,c,ls);
    if(y>mid)upmu(mid+1,r,x,y,c,rs);
    t[p]=(t[ls]+t[rs])%mod;
}
LL query(int l,int r,int x,int y,int p){
    if(x<=l&&y>=r)return t[p];
    int mid=l+r>>1;
    LL re=0;
    pud(l,r,p);
    if(x<=mid)re=(re+query(l,mid,x,y,ls))%mod;
    if(y>mid)re=(re+query(mid+1,r,x,y,rs))%mod;
    return re;
}
int main(){
    scanf("%d%lld",&n,&mod);
    bt(1,n,1);
    scanf("%d",&m);
    int op,x,y,z;
    for(int i=1;i<=m;i++){
        scanf("%d",&op);
        if(op==1){
            scanf("%d%d%d",&x,&y,&z);
            upmu(1,n,x,y,z,1);
        }else if(op==2){
            scanf("%d%d%d",&x,&y,&z);
            upad(1,n,x,y,z,1);
        }else{
            scanf("%d%d",&x,&y);
            printf("%lld\n",query(1,n,x,y,1));
        }
    }
}