【luogu3373】模板 線段樹 2

題面

已知一個數列，你需要進行下面三種操作：
1.將某區間每一個數乘上x
2.將某區間每一個數加上x
3.求出某區間每一個數的和

題解

區間修改+區間查詢。
維護兩個LazyTag

#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 100010;
typedef long long LL;

int n, m;
LL a[maxn],mod;
struct node{
    int l, r;
    LL val, addmark, mulmark;
}sgt[maxn<<2];
void build(int p, int l, int r){
    sgt[p].l = l, sgt[p].r = r;
    sgt[p].mulmark=1, sgt[p].addmark=0;
    if(l == r){
        sgt[p].val = a[l];
    }else{
        int m = (l+r)/2;
        build(p*2,l,m);
        build(p*2+1,m+1,r);
        sgt[p].val = sgt[p*2].val+sgt[p*2+1].val;
    }
    sgt[p].val %= mod;
}
void pushdown(int p){
    if(sgt[p].addmark==0&&sgt[p].mulmark==1)return ;
    //初始化父節點
    LL t1 = sgt[p].addmark, t2 = sgt[p].mulmark;
    sgt[p].addmark = 0, sgt[p].mulmark = 1;
    //維護標記
    sgt[p*2].mulmark = (sgt[p*2].mulmark*t2)%mod;
    sgt[p*2+1].mulmark = (sgt[p*2+1].mulmark*t2)%mod;
    sgt[p*2].addmark = (sgt[p*2].addmark*t2+t1)%mod;
    sgt[p*2+1].addmark = (sgt[p*2+1].addmark*t2+t1)%mod;
    //更新當前值,我們規定乘法優先更新(加法優先會損失精度)
    int l = sgt[p].l, r = sgt[p].r, m = (l+r)/2;
    sgt[p*2].val=(sgt[p*2].val*t2+t1*(m-l+1))%mod;//先乘以乘法標記再加上已用乘法標記更新過的加法標記。
    sgt[p*2+1].val=(sgt[p*2+1].val*t2+t1*(r-m))%mod;
}
void add(int p, int l, int r, LL v){
    if(l <= sgt[p].l && sgt[p].r <= r){
        sgt[p].val = (sgt[p].val+(sgt[p].r-sgt[p].l+1)*v)%mod;
        sgt[p].addmark = (sgt[p].addmark+v)%mod;
        return ;
    }
    pushdown(p);
    int m = (sgt[p].l+sgt[p].r)/2;
    if(l <= m)add(p*2,l,r,v);
    if(r > m)add(p*2+1,l,r,v);
    sgt[p].val = (sgt[p*2].val+sgt[p*2+1].val)%mod;
}
void times(int p, int l, int r, LL v){
    if(l <= sgt[p].l && sgt[p].r <= r){
        sgt[p].val = (sgt[p].val*v)%mod;
        sgt[p].mulmark = (sgt[p].mulmark*v)%mod;
        sgt[p].addmark = (sgt[p].addmark*v)%mod;//原先的加法標記也要乘
        return ;
    }
    pushdown(p);
    int m = (sgt[p].l+sgt[p].r)/2;
    if(l <= m)times(p*2,l,r,v);
    if(r > m)times(p*2+1,l,r,v);
    sgt[p].val = (sgt[p*2].val+sgt[p*2+1].val)%mod;
}
LL query(int p, int l, int r){
    if(l <= sgt[p].l && sgt[p].r <= r)return sgt[p].val;
    pushdown(p); //pushdown
    LL m = (sgt[p].l+sgt[p].r)/2, ans = 0;
    if(l <= m)ans += query(p*2,l,r);
    if(r > m)ans += query(p*2+1,l,r);
    return ans%mod;
}

int main(){
    ios::sync_with_stdio(false);
    cin>>n>>m>>mod;
    for(int i = 1; i <= n; i++)cin>>a[i];
    build(1,1,n);
    for(int i = 1; i <= m; i++){
        int op;  cin>>op;
        if(op == 1){
            LL x, y, z;  cin>>x>>y>>z;
            times(1,x,y,z);
        }else if(op == 2){
            LL x, y, z;  cin>>x>>y>>z;
            add(1,x,y,z);
        }else{
            LL x, y;  cin>>x>>y;
            cout<<query(1,x,y)%mod<<"\n";
        }
    }
    return 0;
}

寫的時候偷懶直接複製加法函式到乘法然後改。。。結果忘記改遞迴的地方了，調了半個多小時調不出來。