树状数组详解

今天被leetcode的每日一题教育了，涉及到知识盲区了。没学过线段树和树状数组。
附上题目链接https://leetcode-cn.com/problems/range-sum-query-mutable/

这是一道很经典的题目，通常还能拓展出一大类问题。

针对不同的题目，我们有不同的方案可以选择（假设我们有一个数组）：
数组不变，求区间和：「前缀和」、「树状数组」、「线段树」
多次修改某个数（单点），求区间和：「树状数组」、「线段树」
多次修改某个区间，输出最终结果：「差分」
多次修改某个区间，求区间和：「线段树」、「树状数组」（看修改区间范围大小）
多次将某个区间变成同一个数，求区间和：「线段树」、「树状数组」（看修改区间范围大小）
总结一下，我们应该按这样的优先级进行考虑：

简单求区间和，用「前缀和」
多次将某个区间变成同一个数，用「线段树」
其他情况，用「树状数组」

先来学习下树状数组
树状数组最精髓的一句话：树状数组本质是二进制规律的应用

树状数组就是用数组模拟树形结构

红色方块则是树状数组，存储的则是子节点值的和：
C[1] = A[1];
C[2] = A[1] + A[2];
C[3] = A[3];
C[4] = A[1] + A[2] + A[3] + A[4];
C[5] = A[5];
C[6] = A[5] + A[6];
C[7] = A[7];
C[8] = A[1] + A[2] + A[3] + A[4] + A[5] + A[6] + A[7] + A[8];

可以发现，这是有规律的
C[i] = A[i - 2^k+1] + A[i - 2^k+2] + ... + A[i];   

/*其中 k 是将 i 表示成二进制以后，从右向左数，遇到 1 则停止时，数出的 0  的个数。
通过 lowbit 高效计算 2^k */

int lowbit(int x){
     return x&(-x);
}
lowbit(34) = 2;//因为34转二进制 1000​​​​​​​10 从最右边往左数第一个‘1’出现在2^1的位置

建议大家多在稿纸上写几个具体的例子来计算 lowbit(i)，进而理解为什么lowbit(i) = 2^k;

给上树状数组的模板

int n;
int a[1005],c[1005]; //对应原数组和树状数组

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

void updata(int i,int k){    //在i位置加上k
    while(i <= n){
        c[i] += k;
        i += lowbit(i);
    }
}

int getsum(int i){        //求A[1 - i]的和
    int res = 0;
    while(i > 0){
        res += c[i];
        i -= lowbit(i);
    }
    return res;
}

想要更深入的理解树状数组还需要多加练习，可以多去leetcode多找点同类型的题目。