[lnsyoj281/luoguP2023/AHOI2009]维护序列

题意

原题链接
给定序列\(a\),要求维护区间加,区间乘,区间查询三种操作

sol

显然线段树,事实上,这是一道板子题(luoguP3373),但由于蒟蒻实在是太蒻了,并没有打过这道题。

区间加

如果我们将区间里的每一个元素都插入线段树做一次修改操作,那么一次修改操作的时间复杂度为\(O(n \log n)\),此时需要引入懒标记(lazytag)
当我们更新/查询到一个区间时,若这个区间的所有元素都在我们要更新/查询的区间中,即被其包含,那么我们就无需继续向下递归,而是在这个节点上记录一个lazytag。当下一次需要更新/查询其儿子时,再将lazytag向下传递
此时我们只需要处理lazytag即可
对于以下序列$$a_1, a_2, a_3, \cdots , a_n$$
此时这个序列的和为$$\sum_{i=1}^{n} a_i$$
此时,若将其中\([l, r]\)区间增加\(c\),则序列变为

\[a_1, a_2, a_3, \cdots , a_l + c, a_{l+1} + c, \cdots , a_{r} + c, \cdots, a_n \]

此时序列和变为$$\sum_{i=1}^{n} a_i + c(r - l + 1)$$
考虑到\(r-l+1\)可以处理出来,所以只需传递\(c\)即可

区间乘

类似于区间加,我们也可以使用lazytag进行操作
记两个lazytag,分别为\(lazymul\)\(lazyadd\),来处理加法和乘法的情况
若将刚才的序列中的\([l, r]\)区间乘\(v\),则序列变为

\[a_1, a_2, a_3, \cdots , v(a_l + c), v(a_{l+1} + c), \cdots , v(a_{r} + c), \cdots, a_n \]

此时序列和变为$$\sum_{1\le i \le n, i \notin [l, r]}a_i + v\cdot \sum_{l\le i \le r} a_i + v\cdot c(r - l + 1)$$
我们发现,当执行区间乘操作时,不仅\(lazymul\)需要修改,而且\(lazyadd\)也需要乘以相应的数
最终,该区间的的和即为\(tr \cdot lazymul + lazyadd\)

代码

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;
typedef long long LL;

const int N = 100005;

LL tree[N * 4];
LL lazyadd[N * 4], lazymul[N * 4];
int a[N];
int n, m, p;

void pushup(int u){
    tree[u] = tree[u << 1] + tree[u << 1 | 1];
}

void pushdown(int u, int l, int r){
    if (l != r){
        lazymul[u << 1] = (LL) lazymul[u << 1] * lazymul[u] % p;
        lazymul[u << 1 | 1] = (LL) lazymul[u << 1 | 1] * lazymul[u] % p;
        lazyadd[u << 1] = ((LL) lazyadd[u << 1] * lazymul[u] % p + lazyadd[u]) % p;
        lazyadd[u << 1 | 1] = ((LL) lazyadd[u << 1 | 1] * lazymul[u] % p + lazyadd[u]) % p; 
        int mid = l + r >> 1;
        tree[u << 1] = ((LL) tree[u << 1] * lazymul[u] % p + lazyadd[u] * (mid - l + 1)) % p;
        tree[u << 1 | 1] = ((LL) tree[u << 1 | 1] * lazymul[u] % p + lazyadd[u] * (r - mid)) % p;
    }
    lazymul[u] = 1, lazyadd[u] = 0;
}

void build(int u, int l, int r){
    lazymul[u] = 1, lazyadd[u] = 0;
    if (l == r) {
        tree[u] = a[l];
        return ;
    }
    int mid = l + r >> 1;
    build(u << 1, l, mid);
    build(u << 1 | 1, mid + 1, r);
    pushup(u);
}

void update(int u, int l, int r, int L, int R, LL val, int op){
    if (L <= l && r <= R){
        if (op) {
            lazyadd[u] = (lazyadd[u] + val) % p;
            tree[u] = (tree[u] + val * (r - l + 1)) % p;
        }
        else {
            lazymul[u] = (LL) lazymul[u] * val % p;
            lazyadd[u] = (LL) lazyadd[u] * val % p;
            tree[u] = (LL) tree[u] * val % p;
        }
        return ;
    }
    pushdown(u, l, r);
    int mid = l + r >> 1;
    if (L <= mid) update(u << 1, l, mid, L, R, val, op);
    if (R > mid) update(u << 1 | 1, mid + 1, r, L, R, val, op);
    pushup(u);
}

LL query(int u, int l, int r, int L, int R){
    if (L <= l && r <= R) return tree[u];
    pushdown(u, l, r);
    LL res = 0, mid = l + r >> 1;
    if (L <= mid) res = (res + query(u << 1, l, mid, L, R)) % p;
    if (R > mid) res = (res + query(u << 1 | 1, mid + 1, r, L, R)) % p;

    return res;
}

int main(){
    scanf("%d%d", &n, &p);
    for (int i = 1; i <= n; i ++ ) scanf("%d", &a[i]);
    build(1, 1, n);

    scanf("%d", &m);
    while (m -- ){
        int op;
        int t, g, c;
        scanf("%d", &op);
        if (op == 1){
            scanf("%d%d%d", &t, &g, &c);
            update(1, 1, n, t, g, c, 0);
        }
        else if (op == 2){
            scanf("%d%d%d", &t, &g, &c);
            update(1, 1, n, t, g, c, 1);
        }
        else {
            scanf("%d%d", &t, &g);
            printf("%lld\n", query(1, 1, n, t, g));
        }
    }

    return 0;
}

蒟蒻犯下的一些若至错误

  1. 建树的时候把a[l]写成了l,结果1h没调出来
  2. 取模没取尽,导致最终炸int
posted @ 2024-05-18 15:00  是一只小蒟蒻呀  阅读(35)  评论(0编辑  收藏  举报