动态求连续区间和(树状数组)（包含原理的保姆级教程）

动态求连续区间和问题

前言

对于树状数组问题，网上很多小伙伴的解释都很到位，但是对于小白而言可能有些跳跃，这里我会从新手接触的角度去介绍 树状数组的解法，有还未解释到的地方希望大家在评论区留言

问题详情

给定 n个数组成的一个数列，规定有两种操作，一是修改某个元素，二是求子数列 [a,b] 的连续和。

输入格式
第一行包含两个整数 n 和 m，分别表示数的个数和操作次数。

第二行包含 n 个整数，表示完整数列。

接下来 m 行，每行包含三个整数 k,a,b （k=0，表示求子数列[a,b] 的和；k=1，表示第 a个数加 b【暗示这里的下标要从 1 开始】）。

数列从 1 开始计数。

输出格式
输出若干行数字，表示 k=0 时，对应的子数列 [a,b]的连续和。

数据范围
1≤n≤100000
1≤m≤100000
1≤a≤b≤n
数据保证在任何时候，数列中所有元素之和均在 int 范围内。

输入样例：
10 5
1 2 3 4 5 6 7 8 9 10
1 1 5
0 1 3
0 4 8
1 7 5
0 4 8
输出样例：
11
30
35

问题分析

要实现的目标

1.实现对数组前缀和求解

要求解从 下标为 a 到下标为 b 的和，如果用前缀和可以使时间复杂度从 O(N) 降到
O(1);

2.实现对数组上每个数的修改，并且每次的修改都会影响到前缀和

常规暴力思路

1. 输入原数组 a [ N ] ,并且计算其前缀和 s [ N ]
2. 对每次查询的数据分类
3. 如果 k == 1 那么直接对数组进行修改，重新计算该位置的数涉及到的前缀和
4. 如果 k==0 那么直接用 s [ right ] - s[ left - 1] 得出区间的总和

树状数组解法

树状数组的定义

首先我们来认识一下树状数组是啥玩意，看图

假设我们要求前 3 个数的前缀和 ，我们只需要图中橙色标记的数字

假设我们要求前 5 个数的前缀和，我们只需要图中绿色标记的数字

到这里细心的小伙伴可能会发现数值为 17 的数字用不到，如果进行前 6，7,8，9 个数的前缀和计算，我们同样用不到 数值为 17 的数字，这说明有的数字始终不会被使用可以去掉，这里的规律是偶数位的数字可以去掉
看图中用黑线划去的，注意这里包含原数组呦。

那么剩下的数字总共恰好有 n 个数，将这 n 个数组成的数组我们叫做树状数组

正规的图是这样的，图中的 t [ N ] 数组即是线段数组

树状数组的功能原理

lowerbit函数

lowbit 函数是树状数组中的一个重要函数，用于计算一个数的最低位的1所对应的值。下面是 lowbit 函数的C++实现及详解：

int lowbit(int x) {
    return x & -x;
}

lowbit 函数的实现非常简洁，它利用了位运算的性质来快速计算一个数的最低位的1所对应的值。

首先，我们需要了解位运算中的与运算（&）和负号运算（-）的特性。

1. 与运算（&）：对于两个二进制数的对应位，只有当两个位都为1时，结果位才为1，否则为0。例如，1010 & 1101 = 1000。

2. 负号运算（-）：对于一个整数 x，-x 可以通过对 x 进行取反（按位取反，即将所有位的0变为1，将所有位的1变为0）后加1来得到。例如，-5 = ~5 + 1 = -6。

在 lowbit 函数的实现中，我们使用了负号运算来计算一个数的补码（即取反加一），然后再与原数进行与运算。

具体来说，我们可以观察到一个数的补码与原数的二进制表示中最低位的1所对应的值是一样的。例如，5 的二进制表示是 101，它的补码是 -5 的二进制表示是 1111 1011，两者的最低位的1所对应的值都是 001，即 1。

因此，通过将一个数与其负号运算的结果进行与运算，我们可以得到它的最低位的1所对应的值。

例如，lowbit(5) 的计算过程如下：

1. 5 的二进制表示是 101。
2. -5 的二进制表示是 1111 1011。
3. 5 & -5 的结果是 0001，即 1。

lowbit 函数的返回值就是一个数的最低位的1所对应的值。

在树状数组中，lowbit 函数常用于计算当前位置 i 的父节点位置 i - lowbit(i)，或者计算当前位置 i 的下一个位置 i + lowbit(i)。

规律找寻

每一行的lowerbit 数值都是相同的；

实现 树状数组 add 函数【在下标为 x 处增加 v】

原理

因为要修改一个数的值，必然要维护整个树形结构的状态
假设我们要修改 a5 在 a5 上面 + 3，那么就要修改对应的 t6 , t8;

由此我们推导出公式

void add(int x,int v){
    for(int i=x;i<=n;i+=lowerbit(i)){
        tr[i]+=v;
    }
}

实现树状数组query操作

举例

假设我们需要求解前 8 个数的前缀和，那么我们可以看出
t8 = a8 + t7 + t6 + t4
7 = 8 - 2^0;
6 = 8 - 2^1;
4 = 8 - 2^2;

那么有规律如下
找到要求的前缀和，向下求索，方式是 x- lowerbit(x)

int query(int x){
    int res=0;
    for(int i=x;i>=1;i-=lowerbit(i)){
        res+=tr[i];
    }
    return res;
}

代码

#include<iostream>
using namespace std;
const int N =1e5 + 7;
int q[N],tr[N],n,m;//这里的tr[N] 代指树状数组，q [ N ]指的是原数组
int lowerbit(int x){
    return x & -x;
}

void add(int x,int v){
    for(int i=x;i<=n;i+=lowerbit(i)){
        tr[i]+=v;
    }
}

int query(int x){
    int res=0;
    for(int i=x;i>=1;i-=lowerbit(i)){
        res+=tr[i];
    }
    return res;
}

int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        cin>>q[i];
        add(i,q[i]);
    }
    while(m--){
        int k,a,b;
        cin>>k>>a>>b;
        if(k==1){
            add(a,b);
        }
        else{
            cout<<query(b)-query(a-1)<<endl;
        }
    }
    return 0;
}