树状数组之区间修改，单点查询

树状数组之区间修改，单点查询

原理

首先我们开三个数组：

• 原数组:a [i], 每一个元素的值
• 差分数组:b [i], 下标 i 的元素存放 a [i]-a [i-1] 的差值，b [1]=a [1]
• 树状数组:c [i], 针对数组 b 行成的树状数组

我们就形成了下面的这种数据结构：

我们发现对于树状数组 C[i]C[i] 来说，query(i)=a[i]query(i)=a[i], 也就是对应数的值.
如果我们想 2->4 的数加 10 怎么办？我们只要修改 b[i] 数组如下:

• b[2]+10b[2]+10
• b[4+1]−10

我们就形成了下面的数据；

我们不难发现差分数组的前缀和就是对应原数组的值即：
getsum(i) = a[i];

核心操作

   下列操作与之前的模板代码有所不同，这里的代码用到父节点和前管辖域的下标来实现找寻下标，其本质一样没有区别，但这样写易懂

int n, m;
int c[N+5]; // c[i] = a[i] - a[i-1]，注意这里多开5个防止数据溢出

inline int lowbit(int x) { return x & -x; }
inline int fa(int p) { return p + lowbit(p); }
inline int left(int p) { return p - lowbit(p); }

void change(int p, int v) {
    for (; p <= N; p = fa(p))
        c[p] += v;
}
void change_range(int l, int r, int v) { //区间修改
    change(l, v); //左节点需改变
    change(r + 1, -v); //右节点的下一个节点需要改变，这里容易发生数据溢出请注意
}
int getsum(int p) { //前缀合,单点查询
    int sum = 0;
    for (; p > 0; p = left(p)) sum += c[p];
    return sum;
}

例题

luogu P3368 【模板】树状数组 2

思路： 简单的树状数组的应用

#include <iostream>
#include <string.h>
using namespace std;

typedef long long ll;
const int maxn = 1e6 + 5, maxe = 1e6 + 5; //点与边的数量
const int N = maxn;

int n, m;
int c[N]; // c[i] = a[i] - a[i-1]

inline int lowbit(int x) { return x & -x; }
inline int fa(int p) { return p + lowbit(p); }
inline int left(int p) { return p - lowbit(p); }

void change(int p, int v) {
    for (; p <= N; p = fa(p))
        c[p] += v;
}

inline void change_range(int l, int r, int v) {
    change(l, v);
    change(r + 1, -v);
}

int getsum(int p) { //前缀合,单点查询
    int sum = 0;
    for (; p > 0; p = left(p)) sum += c[p];
    return sum;
}

int main() {
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; ++i) {
        int t;
        scanf("%d", &t);
        change_range(i, i, t);//初始化树状数组，区间i~i就是i
    }
    for (int i = 1; i <= m; ++i) {
        int o, x, y, z;
        scanf("%d", &o);
        if (o == 1) {
            scanf("%d%d%d", &x, &y, &z);
            change_range(x, y, z);
        }
        else {
            scanf("%d", &x);
            ll ans = getsum(x);
            printf("%lld\n", ans);
        }
    }
    return 0;
}