树状数组总结(C/C++)

1.树状数组分析

 1.1 树状数组简单解析
 核心：利用二进制末尾1的位置进行分类构建每一层，然后可以发现如果如下图进行分层的话，下面的积木想网上方走只需要增加末尾二进制1，往左边走只需要减少末尾二进制1即可，可以利用这个便利性构建树状数组。

 1.2 适用范围
 A. 单点修改 + 区间查询 = (核心本质)
 毫无疑问的，我们可以快速修改到具体的点的位置，然后不断上移修改包含它的区间(完成单点修改)，利用找到具体点然后去不断左移找到前方所有的区间的和(完成整体区间和查询)，查询某个具体中间区间(用两个大区间相减即可)。
 B. 区间修改 + 单点查询
 其实这里本质还是使用了上方的单点修改+区间查询本质。只不过我们初始化存储的时候存储的是差分数组，当对差分数组进行单点修改也就等效于对整个后面进行区间修改，对差分数组进行求和(等效于进行单点查询)。
 C. 本质 = 前缀和 / 差分操作
 很难同时完成区间的修改和区间查询操作，同时区间的修改与查询通常还是会适用线段树进行完成。

2.树状数组实现

 2.1 方块之间移动实现(核心): Lowbit()函数
 正数和负数二进制表示出来，恰好只有在x末尾1的位置上才有两个1进行&操作可以保留下来.(自己画个图看看就知道了)

int lowbit(int x){ //得到二进制x末尾1的位置大小
    return x & (-x);
}

 2.2 方块区间上移(修改所有父区间)——通常对应数组的单点修改
 合法范围内，不断加上lowvbit(x)即可向上走到更高区间了

for(int i = x; i <= n; i += lowbit(i)){ // 从 t[x]开始不断往右走
	t[i] += y; // 通常都是修改, + y 
}

 2.3 方块区间左移(查找所有左边区间即前面所有区间)——通常对应数组的区间求和，求解所有左边区间的和
 同理，合法范围内，不断减去lowvbit(x)即可向左走到前方区间

for(int i = x; i >= 1; i -= lowbit(i)) 
	res += t[i]; // 通常都是求和, 用res不断加

3.树状数组代码运行

 3.1 单点修改+区间查询

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long LL;
const int N = 5;
int a[N] = {0, 1, 2, 3, 4}, t[N], n = 4;
int lowbit(int x){
    return x & -x;
}
void add(int x, int k){ //沿着连线更新所有x的父节点
    for(int i = x; i <= n; i += lowbit(i)) t[i] += k;
}
int search(int x){ // 查找 1 - x 之间所有数字的和 , 求部分区间相减去即可
    LL sum = 0;
    for(int i = x; i >= 1; i -= lowbit(i)) sum += t[i];
    return sum;
}
int main(){
    for(int i = 1; i <= 4; i ++ ){ //注意存储必须从下标1开始, 0下标+ lowbit()不会上移
        add(i, a[i]); // 初始化放入t[]当中
    }
    cout << "需要处理的数组为 1 2 3 4 " << endl;
    cout << "展示求和功能 1 - 4 的和为：" << search(4) << endl;
    cout << "展示求和功能 2 - 3 的和为：" << search(3) - search(1) << endl;

    add(3, 10);
    cout << "对第3个数字 + 10" << endl;
    cout << "第3个数字更新后的结果为：" << search(3) - search(2) << endl;

}

 3.2 区间修改+单点查询 (差分数组)
 存储的时候一定要存储数组差分的结果，这样子之后的单点修改就会对应区间修改，单点查询就对应区间求和

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long LL;
const int N = 5;
int a[N] = {0, 1, 2, 3, 4}, t[N], n = 4;
int lowbit(int x){
    return x & -x;
}
void add(int x, int k){ //沿着连线更新所有x的父节点
    for(int i = x; i <= n; i += lowbit(i)) t[i] += k;
}
int search(int x){ // 查找 1 - x 之间所有数字的和 ,即等效于单点查询(差分求和)
    LL sum = 0;
    for(int i = x; i >= 1; i -= lowbit(i)) sum += t[i];
    return sum;
}
int main(){
    for(int i = 1; i <= 4; i ++ ){
        add(i, a[i] - a[i - 1]); // 差分数组 初始化放入t[]当中
    }
    cout << "需要处理的数组为 1 2 3 4 " << endl;
    add(2, 10), add(4, -10); // 对应于差分的区间修改
    cout << "将区间位置 2 - 3上面数字 + 10" << endl;
    cout << "处理完成后的结果为 ";
    for(int i = 1; i <= 4; i ++ ){
        cout << search(i) << " ";
    }
    cout << '\n';

}