洛谷P1438 无聊的数列

最新推荐文章于 2025-01-15 16:30:28 发布

JXNU_SONG

最新推荐文章于 2025-01-15 16:30:28 发布

阅读量228

点赞数

分类专栏：数据结构

本文链接：https://blog.csdn.net/qq_45863710/article/details/119422323

版权

数据结构专栏收录该内容

35 篇文章 1 订阅

订阅专栏

无聊的数列

题目链接

题目大意：给出一个有 $n$ 个元素的序列 $a$ ，和 $m$ 次操作。倘若是修改操作，给出 $l,r,a_1,d$ ，意为在 $[l, r]$ 这个区间内每个元素加上一个等差数列的元素。询问操作则询问某一个元素的值。

对于这种区间加一个有规律的有序值的问题通常将其转化为与其下标有关，我们假设某个元素的下标为 $i d x$ ，那么对于一次修改操作，其值就会加上 $a_1+(idx-l)*d$ ，拆开： $a_1-l*d)+idx*d$ ，那么假设一个区间 $[l, r]$ ，修改一次整个区间就会加上： $r-l+1)*(a_1-l*d)+(sum[r]-sum[l-1])*d$ sum数组代表等差数列前缀和。那么我们就可以利用线段树去做这个题了，我们维护两个 $l a z y$ 标记即可，一个就是正常的区间加的标记，另一个就是维护其区间的下标和被加过多少次的 $l a z y$ 。

#include <bits/stdc++.h>
#define lson rt<<1
#define rson rt<<1|1

using namespace std;

typedef long long ll;

const int N = 1e5 + 5;

int n, m, a[N];

int op, idx;
ll a1, d, ul, ur;

ll tree[N << 2], lz1[N << 2], lz2[N << 2], s[N];

void init() {
    s[1] = 1;
    for (int i = 2; i <= n; ++i) {
        s[i] = s[i - 1] + i;
    }
}

void push_up(int rt) { tree[rt] = tree[lson] + tree[rson]; }

void build(int rt, int l, int r) {
    if (l == r) {
        scanf("%lld", &tree[rt]);
        return ;
    }
    int mid = l + r >> 1;
    build(lson, l, mid);
    build(rson, mid + 1, r);
    push_up(rt);
}

void push_down(int rt, int l, int r) {
    if (!lz1[rt]) return ;
    int mid = l + r >> 1;
    lz1[lson] += lz1[rt]; lz2[lson] += lz2[rt];
    lz1[rson] += lz1[rt]; lz2[rson] += lz2[rt];
    tree[lson] += (mid - l + 1) * lz1[rt] + (s[mid] - s[l - 1]) * lz2[rt];
    tree[rson] += (r - mid) * lz1[rt] + (s[r] - s[mid]) * lz2[rt];
    lz1[rt] = lz2[rt] = 0;
}

void update(int rt, int l, int r, int L, int R) {
    if (L <= l && r <= R) {
        tree[rt] += (a1 - ul * d) * (r - l + 1) + (s[r] - s[l - 1]) * d;
        lz1[rt] += a1 - ul * d;
        lz2[rt] += d;
        return ;
    }
    push_down(rt, l, r);
    int mid = l + r >> 1;
    if (mid >= L) update(lson, l, mid, L , R);
    if (mid < R) update(rson, mid + 1, r, L, R);
    push_up(rt);
}

ll query(int rt, int l, int r) {
    if (l == r) return tree[rt];
    int mid = l + r >> 1;
    push_down(rt, l, r);
    if (idx <= mid) return query(lson, l, mid);
    return query(rson, mid + 1, r); 
}

int main() {
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif
    scanf("%d%d", &n, &m);
    init();
    build(1, 1, n);
    while(m--) {
        scanf("%d", &op);
        if (op == 1) {
            scanf("%lld%lld%lld%lld", &ul, &ur, &a1, &d);
            update(1, 1, n, ul, ur);
        }
        else {
            scanf("%d", &idx);
            printf("%lld\n", query(1, 1, n));
        }
    }
}