模板 - 树状数组

怎甘沉沦 提交于 2019-11-30 12:26:28

区间加值,区间求和的树状数组,通过差分来实现。

并没有线段树那么通用。

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

ll bit[100005];
ll bit2[100005];

int n;

void add(int x, ll v) {
    for(; x <= n; x += x & -x)
        bit[x] += v;
}

ll sum(int x) {
    ll res = 0;
    for(; x; x -= x & -x)
        res += bit[x];
    return res;
}

void add2(int x, ll v) {
    for(; x <= n; x += x & -x)
        bit2[x] += v;
}

ll sum2(int x) {
    ll res = 0;
    for(; x; x -= x & -x)
        res += bit2[x];
    return res;
}

ll ans(int n) {
    return 1ll * (n + 1) * sum(n) - sum2(n);
}

int main() {
#ifdef Yinku
    freopen("Yinku.in", "r", stdin);
#endif // Yinku
    int m;
    scanf("%d%d", &n, &m);

    for(int i = 1, x, px = 0; i <= n; ++i) {
        scanf("%d", &x);
        add(i, x - px);
        add2(i, 1ll * i * (x - px));
        px = x;
    }

    for(int i = 1; i <= m; ++i) {
        char op[5];
        scanf("%s", op);
        if(op[0] == 'C') {
            int l, r, d;
            scanf("%d%d%d", &l, &r, &d);
            add(l, d);
            add2(l, l * d);
            add(r + 1, -d);
            add2(r + 1, 1ll * (r + 1) * -d);
        } else {
            int l, r;
            scanf("%d%d", &l, &r);
            printf("%lld\n", ans(r) - ans(l - 1));
        }
    }
}
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!