【悬关】70pts求调
查看原帖
【悬关】70pts求调
452893
TriQUery楼主2024/9/6 20:46

悬关,以下蒟蒻代码

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

const int NR = 1e5 + 10;
struct Node {
    int l, r;
    long long sum;
} st[NR << 2];
long long lazy[NR << 2];
long long mul[NR << 2];
long long a[NR], n, m, mod;

int ls(int id) { return id * 2; }
int rs(int id) { return id * 2 + 1; }
int len(int id) { return st[id].r - st[id].l + 1; }
void pushup(int id) { st[id].sum = (st[ls(id)].sum + st[rs(id)].sum) % mod; }

void pushdown(int id) {
    int l = ls(id), r = rs(id);
    long long k1 = lazy[id], k2 = mul[id];
    
    st[l].sum = (st[l].sum * k2 % mod + len(l) * k1 % mod) % mod;
    st[r].sum = (st[r].sum * k2 % mod + len(r) * k1 % mod) % mod;
    
    lazy[l] = (lazy[l] * k2 % mod + k1) % mod;
    lazy[r] = (lazy[r] * k2 % mod + k1) % mod;
    
    mul[l] = mul[l] * k2 % mod;
    mul[r] = mul[r] * k2 % mod;
    
    lazy[id] = 0; mul[id] = 1;
}

void build(int id, int l, int r) {
    if (l == r) {
        st[id] = {l, r, a[l]};
        return ;
    }
    
    int mid = l + r >> 1;
    build(ls(id), l, mid); build(rs(id), mid + 1, r);
    
    st[id] = {l, r, 0};
    pushup(id);
}

void add(int id, int l, int r, long long k) {
    if (r < st[id].l || st[id].r < l) {
        return ;
    }
    if (l <= st[id].l && st[id].r <= r) {
        st[id].sum = (st[id].sum + (st[id].r - st[id].l + 1) * k) % mod;
        lazy[id] = (lazy[id] + k) % mod;
        return ;
    }
    
    pushdown(id);
    add(ls(id), l, r, k); add(rs(id), l, r, k);
    pushup(id);
}

void addmul(int id, int l, int r, long long k) {
    if (r < st[id].l || st[id].r < l)
        return ;
    if (l <= st[id].l && st[id].r <= r) {
        st[id].sum = st[id].sum * k % mod;
        lazy[id] = lazy[id] * k % mod;
        mul[id] = mul[id] * k % mod;
        return ;
    }
    
    pushdown(id);
    addmul(ls(id), l, r, k); addmul(rs(id), l, r, k);
    pushup(id);
}

long long TriQUery(int id, int l, int r) {
    if (r < st[id].l || st[id].r < l)
        return 0ll;
    if (l <= st[id].l && st[id].r <= r)
        return st[id].sum % mod;
    
    pushdown(id);
    return (TriQUery(ls(id), l, r) + TriQUery(rs(id), l, r)) % mod;
}

void print(int id, int l, int r, int k) {
    for (int i = 1; i <= k; i++)
        cout << "    ";
    cout << id << ": " << l << " ~ " << r << " " << st[id].sum << endl;
    for (int i = 1; i <= k; i++)
        cout << "    ";
    cout << lazy[id] << " " << mul[id] << endl;
    
    if (l == r) return ;
    int mid = l + r >> 1;
    print(ls(id), l, mid, k + 1);
    print(rs(id), mid + 1, r, k + 1);
}

int main() {
    for (int i = 0; i < NR; i++)
        mul[i] = 1;
    
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    
    cin >> n >> m >> mod;
    for (int i = 1; i <= n; i++)
        cin >> a[i], a[i] %= mod;
    build(1, 1, n);
    
    for (int i = 1; i <= m; i++) {
        // print(1, 1, n, 0);
        int op, x, y, k;
        cin >> op >> x >> y;
        if (op == 1) {
            cin >> k;
            addmul(1, x, y, k % mod);
        }
        if (op == 2) {
            cin >> k;
            add(1, x, y, k % mod);
        }
        if (op == 3) cout << TriQUery(1, x, y) % mod << endl;
    }
    return 0;
}
2024/9/6 20:46
加载中...