为什么有三个过不了
查看原帖
为什么有三个过不了
1461134
jiayuhang楼主2024/11/20 21:38
#include <bits/stdc++.h>
#define long long int
#define lc (p << 1)
#define rc (p << 1 | 1)
#define N 200005
using namespace std;

struct node {
    int l, r, sum, add;
} tr[N * 4];

int w[N];
int m, n;

// 更新父节点的值
void pushup(int p) {
    tr[p].sum = tr[lc].sum + tr[rc].sum;
}

// 向下传递更新值
void pushdown(int p) {
    if (tr[p].add) {
        // 更新子节点的sum值
        tr[lc].sum += (tr[lc].r - tr[lc].l + 1) * tr[p].add;
        tr[rc].sum += (tr[rc].r - tr[rc].l + 1) * tr[p].add;

        // 将add值传递给子节点
        tr[lc].add += tr[p].add;
        tr[rc].add += tr[p].add;

        tr[p].add = 0;  // 清空当前节点的add值
    }
}

// 建树
void build(int p, int l, int r) {
    tr[p] = {l,r,w[l],0};

    if (l == r) {
        return;
    }

    int m = (l + r) >> 1;
    build(lc, l, m);
    build(rc, m + 1, r);
    pushup(p);  // 合并结果
}

// 区间更新:给区间[x, y]加上k
void update(int p, int x, int y, int k) {
    if (x <= tr[p].l && tr[p].r <= y) {
        tr[p].sum += (tr[p].r - tr[p].l + 1) * k;
        tr[p].add += k;
        return;
    }

    pushdown(p);  // 需要先pushdown
    int m = (tr[p].l + tr[p].r) >> 1;
    if (x <= m) update(lc, x, y, k);
    if (y > m) update(rc, x, y, k);
    pushup(p);
}

// 区间查询:查询区间[x, y]的和
int query(int p, int x, int y) {
    if (x <= tr[p].l && tr[p].r <= y) {
        return tr[p].sum;
    }

    pushdown(p);  // 需要先pushdown
    int sum = 0;
    int m = (tr[p].l + tr[p].r) >> 1;
    if (x <= m) sum += query(lc, x, y);
    if (y > m) sum += query(rc, x, y);
    return sum;
}

signed main() {
    scanf("%d%d", &m, &n);
    for (int i = 1; i <= m; i++) {
        scanf("%d", &w[i]);
    }
    int op, x, y, k;
    build(1, 1, m);  // 初始化线段树

    while (n--) {
        scanf("%d%d%d", &op, &x, &y);
        if (op == 1) {
            scanf("%d", &k);
            update(1, x, y, k);  // 更新操作
        } else if (op == 2) {
            int ans = query(1, x, y);  // 查询操作
            printf("%d\n", ans);
        }
    }

    return 0;
}

2024/11/20 21:38
加载中...