struct SegmentTree {
    vector<ll> info, tag, infolen;
    int n;
    SegmentTree(int n_, const vector<int> &init) {
        n = n_;
        info.assign(4 * n + 1, 0);
        infolen.assign(4 * n + 1, 0);
        tag.assign(4 * n + 1, 0);
        build(1, 1, n, init);
    }

    void push_up(int u) {
        info[u] = info[u * 2] + info[u * 2 + 1];
        infolen[u] = infolen[u * 2] + infolen[u * 2 + 1];
    }

    void build(int u, int l, int r, const vector<int> &init) {
        if (l == r) {
            info[u] = init[l];
            infolen[u] = 1;
            return;
        }
        int m = l + r >> 1;
        build(u * 2, l, m, init);
        build(u * 2 + 1, m + 1, r, init);
        push_up(u);
    }

    void aply(int u, ll t) {
        tag[u] += t;
        info[u] += t * infolen[u];
    }

    void push_down(int u) {
        if (tag[u]) {
            aply(u * 2, tag[u]);
            aply(u * 2 + 1, tag[u]);
            tag[u] = 0;
        }
    }

    ll query(int u, int ul, int ur, int l, int r) {
        if (l <= ul && ur <= r) {
            return info[u];
        }
        if (r < ul || l > ur) return 0;
        int m = ul + ur >> 1;
        push_down(u);
        return query(u * 2, ul, m, l, r) + query(u * 2 + 1, m + 1, ur, l, r);
    }

    ll query(int l, int r) {
        return query(1, 1, n, l, r);
    }

    void change(int u, int ul, int ur, int l, int r, ll y) {
        if (l <= ul && ur <= r) {
            aply(u, y);
            return;
        }
        if (r < ul || l > ur) return;
        push_down(u);
        int m = ur + ul >> 1;
        change(u * 2, ul, m, l, r, y);
        change(u * 2 + 1, m + 1, ur, l, r, y);
        push_up(u);
    }

    void change(int l, int r, ll y) {
        change(1, 1, n, l, r, y);
    }
};
void solve() {
    int n, m;
    cin >> n >> m;

    vector<int> a(n + 1);
    for (int i = 1; i <= n; i++) cin >> a[i];

    SegmentTree t(n, a);
    
    while (m--) {
        int o, x, y, v;
        cin >> o >> x >> y;
        if (o == 2) {
            cout << t.query(x, y) << '\n';
        } else {
            cin >> v;
            t.change(x, y, v);
        }
    }
}


0 条评论

目前还没有评论...

信息

ID
435
时间
ms
内存
MiB
难度
6
标签
递交数
37
已通过
12
上传者