洛谷 P3373 【模板】线段树 2

题目传送门

参考：P3373 【模板】线段树 2 题解

对线段树的区间更新又有了新的认识
代码：

#include<bits/stdc++.h>
using namespace std;

#define lson root<<1,l,mid
#define rson root<<1|1,mid+1,r
typedef long long ll;
const int maxn=100000+100;

struct Tree{

    ll v,mul,add;
}tree[maxn<<2];
ll ans[maxn],p;
int n,m;

void build(int root,int l,int r){

    if(l==r){

        tree[root].v=ans[l]%p;
        tree[root].mul=1;
        tree[root].add=0;
        return ;
    }
    tree[root].mul=1;
    tree[root].add=0;
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    tree[root].v=(tree[root<<1].v+tree[root<<1|1].v)%p;
}

void updateUp(int root,int l,int r,int mid){

    tree[root<<1].v=(tree[root<<1].v*tree[root].mul+tree[root].add*(mid-l+1))%p;
    tree[root<<1].mul=(tree[root<<1].mul*tree[root].mul)%p;
    tree[root<<1].add=(tree[root<<1].add*tree[root].mul+tree[root].add)%p;
    tree[root<<1|1].v=(tree[root<<1|1].v*tree[root].mul+tree[root].add*(r-mid))%p;
    tree[root<<1|1].mul=(tree[root<<1|1].mul*tree[root].mul)%p;
    tree[root<<1|1].add=(tree[root<<1|1].add*tree[root].mul+tree[root].add)%p;
    tree[root].mul=1;
    tree[root].add=0;
}

void updateAdd(int root,int l,int r,int L,int R,ll v){

    if(L>r || R<l) return ;
    if(L<=l && R>=r){

        tree[root].v=(tree[root].v+v*(r-l+1))%p;
        tree[root].add=(tree[root].add+v)%p;
        return ;
    }   

    int mid=(l+r)>>1;
    if(tree[root].add || tree[root].mul!=1) updateUp(root,l,r,mid);
    updateAdd(lson,L,R,v);
    updateAdd(rson,L,R,v);
    tree[root].v=(tree[root<<1].v+tree[root<<1|1].v)%p;
}

void updateMul(int root,int l,int r,int L,int R,ll v){

    if(L>r || R<l) return ;
    if(L<=l && R>=r){

        tree[root].v=(tree[root].v*v)%p;
        tree[root].mul=(tree[root].mul*v)%p;
        tree[root].add=(tree[root].add*v)%p;
        return ;
    }   

    int mid=(l+r)>>1;
    if(tree[root].add || tree[root].mul!=1) updateUp(root,l,r,mid);
    updateMul(lson,L,R,v);
    updateMul(rson,L,R,v);
    tree[root].v=(tree[root<<1].v+tree[root<<1|1].v)%p;
}

ll query(int root,int l,int r,int L,int R){

    if(L>r || R<l) return 0;
    if(L<=l && R>=r) return tree[root].v;

    int mid=(l+r)>>1;
    if(tree[root].add || tree[root].mul!=1) updateUp(root,l,r,mid);
    return (query(lson,L,R)+query(rson,L,R))%p;
}

int main(){

    scanf("%d%d%lld",&n,&m,&p);
    for(int i=1;i<=n;i++) scanf("%lld",&ans[i]);
    build(1,1,n);
    while(m--){

        int type,l,r;
        ll v;
        scanf("%d",&type);
        if(type==1){

            scanf("%d%d%lld",&l,&r,&v);
            updateMul(1,1,n,l,r,v);
        }
        else if(type==2){

            scanf("%d%d%lld",&l,&r,&v);
            updateAdd(1,1,n,l,r,v);
        }
        else{

            scanf("%d%d",&l,&r);
            printf("%lld\n",query(1,1,n,l,r));
        }
    }
}