bzoj1798: [Ahoi2009]Seq 维护序列seq 线段树

好久没有写线段树了，所以看到这个题就直接写了。。。

然后发现脑残了。一直TE。。。。。然后想起来只要往下乘的时候都换成ax+b就可以了。。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
#define getmid int mid=(l+r)>>1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define maxn 500000
typedef long long sint;
sint sum[maxn];
sint cal[maxn],add[maxn],mod;
sint getsint()
{
    char c;sint res;
    while(c=getchar(),c<'0'||c>'9');
    res=c-'0';
    while(c=getchar(),c>='0'&&c<='9')
        res=res*10+c-'0';
    return res;
}
int getint()
{
    char c;int res;
    while(c=getchar(),c<'0'||c>'9');
    res=c-'0';
    while(c=getchar(),c>='0'&&c<='9')
        res=res*10+c-'0';
    return res;
}
void pushup(int rt)
{
    sum[rt]=(sum[rt<<1]+sum[rt<<1|1])%mod;
}
void pushdown(int rt,int l,int r)
{
        getmid;
        add[rt<<1]=((add[rt<<1]*cal[rt])+add[rt])%mod;
        add[rt<<1|1]=((add[rt<<1|1]*cal[rt])+add[rt])%mod;
 
        cal[rt<<1]=(cal[rt<<1]*cal[rt])%mod;
        cal[rt<<1|1]=(cal[rt<<1|1]%mod*cal[rt])%mod;
        sum[rt<<1]=(sum[rt<<1]*cal[rt]+add[rt]*(mid-l+1))%mod;
        sum[rt<<1|1]=(sum[rt<<1|1]*cal[rt]+add[rt]*(r-mid))%mod;
 
        add[rt]=0;
        cal[rt]=1;
}
void build(int l,int r,int rt)
{
    getmid;
    cal[rt]=1;
    if(l==r)
    {
        sum[rt]=getint()%mod;
        return;
    }
    build(lson);
    build(rson);
    pushup(rt);
}
void update(int l,int r,int rt,int x,int y,int val)
{
    if(x<=l&&y>=r)
    {
        add[rt]+=val;
        add[rt]%=mod;
        sum[rt]=(sum[rt]+val*(r-l+1))%mod;
        return;
    }
    pushdown(rt,l,r);
    getmid;
    if(x<=mid) update(lson,x,y,val);
    if(y>mid) update(rson,x,y,val);
    pushup(rt);
}
void multi(int l,int r,int rt,int x,int y,int val)
{
    if(x<=l&&y>=r)
    {
        cal[rt]*=val;cal[rt]%=mod;
        add[rt]*=val;add[rt]%=mod;
        sum[rt]*=val;sum[rt]%=mod;
        return;
    }
    pushdown(rt,l,r);
    getmid;
    if(x<=mid) multi(lson,x,y,val);
    if(y>mid) multi(rson,x,y,val);
    pushup(rt);
}
sint query(int l,int r,int rt,int x,int y)
{
    if(x<=l&&y>=r)
    {
        return sum[rt]%mod;
    }
    pushdown(rt,l,r);
    getmid;
    sint tmp=0;
    if(x<=mid) tmp+=query(lson,x,y);
    if(y>mid) tmp+=query(rson,x,y);
    tmp%=mod;
    return tmp;
}
int main()
{
    int n,m;
    scanf("%d%lld",&n,&mod);
    build(1,n,1);
    m=getint();
    int op,a,b;
    sint c;
    while(m--)
    {
        op=getint();
        if(op==1)
        {
            a=getint();
            b=getint();
            c=getsint()%mod;
            multi(1,n,1,a,b,c);
        }
        if(op==2)
        {
            a=getint();
            b=getint();
            c=getsint()%mod;
            update(1,n,1,a,b,c);
        }
        if(op==3)
        {
            a=getint();
            b=getint();
            printf("%lld\n",query(1,n,1,a,b)%mod);
        }
    }
    return 0;
}