【洛谷P3373 线段树2】乘法加法模板线段树

https://www.luogu.org/problem/P3373
这个模板题可真是有点意思。。。
重点在lazytag上，加上乘法操作的tag。
update乘的时候要把加的标记也乘一下！
下放标记的时候也要记得先乘后加！

#include
#include
#define MAXN 1000001
#define ll long long
using namespace std;
unsigned ll n,m,P,a[MAXN],ans[MAXN<<2],tag[MAXN<<2],tag2[MAXN<<2];
inline ll ls(ll x){return x<<1;}
inline ll rs(ll x){return x<<1|1;}
void scan(){
    cin>>n>>m>>P;
    for(ll i=1;i<=n;i++)
    scanf("%lld",&a[i]);
}
inline void push_up(ll p){
    ans[p]=ans[ls(p)]+ans[rs(p)];
    ans[p]%=P;
}
void build(ll p,ll l,ll r){ 
    tag[p]=0;
    tag2[p]=1;
    if(l==r){ans[p]=a[l]%P;return ;}
    ll mid=(l+r)>>1;
    build(ls(p),l,mid);
    build(rs(p),mid+1,r);
    push_up(p);
} 

inline void push_down(ll p,ll l,ll r){
    ll mid=(l+r)>>1;
    ans[ls(p)] = (ans[ls(p)]*tag2[p] + tag[p]*(mid-l+1ll)) %P;
    ans[rs(p)] = (ans[rs(p)]*tag2[p] + tag[p]*(r-mid)) %P;
    tag[ls(p)] = (tag[ls(p)]*tag2[p] + tag[p]) %P;
    tag[rs(p)] = (tag[rs(p)]*tag2[p] + tag[p]) %P;
    tag2[ls(p)]*=tag2[p];
    tag2[ls(p)]%=P;
    tag2[rs(p)]*=tag2[p];
    tag2[rs(p)]%=P;
    tag[p]=0;
    tag2[p]=1;
}

inline void update2(ll nl,ll nr,ll l,ll r,ll p,ll k){
    if(nl<=l&&r<=nr){ 
        ans[p]*=k;
        ans[p]%=P;
        tag[p]*=k;
        tag[p]%=P;
        tag2[p]*=k;
        tag2[p]%=P;
        return ;
    }
    push_down(p,l,r);
    ll mid=(l+r)>>1; 
    if(nl<=mid)update2(nl,nr,l,mid,ls(p),k);
    if(nr>mid) update2(nl,nr,mid+1,r,rs(p),k);
    push_up(p);
}
inline void update(ll nl,ll nr,ll l,ll r,ll p,ll k){
    if(nl<=l&&r<=nr){ 
        ans[p]+=k*(r-l+1);
        ans[p]%=P;
        tag[p]+=k;
        tag[p]%=P;
        return ;
    }
    push_down(p,l,r);
    ll mid=(l+r)>>1;
    if(nl<=mid)update(nl,nr,l,mid,ls(p),k);
    if(nr>mid) update(nl,nr,mid+1,r,rs(p),k);
    push_up(p);
}
ll query(ll q_x,ll q_y,ll l,ll r,ll p){
    ll res=0;
    if(q_x<=l&&r<=q_y)return ans[p]%P;
    ll mid=(l+r)>>1;
    push_down(p,l,r);
    if(q_x<=mid)res+=query(q_x,q_y,l,mid,ls(p));
    res%=P;
    if(q_y>mid) res+=query(q_x,q_y,mid+1,r,rs(p));
    res%=P;
    return res;
}
int main(){
    ll a1,b,c,d,e,f;
    scan();
    //长度为n，m个操作 
    build(1,1,n);
    while(m--){
        scanf("%lld",&a1);
        switch(a1){
            case 1:{
                scanf("%lld%lld%lld",&b,&c,&d);//从b到c，*d
                update2(b,c,1,n,1,d);
                break;
            }
            case 2:{
                scanf("%lld%lld%lld",&b,&c,&d);//从b到c，+d 
                update(b,c,1,n,1,d);
                break;
            }
            case 3:{
                scanf("%lld%lld",&e,&f);
                printf("%lld\n",query(e,f,1,n,1));
                break;
            }
        }
    }
    return 0;
}