树状数组的第n次学习笔记

树状数组能解决什么问题？

树状数组能高效地维护前缀和，尤其是“查区间改点”类型的前缀和。
如果树状数组是c，原数组是a，那么c[i]表示的是a[i]，a[i - lowbit(i)]…这些数的和。
树状数组能直接求的是1~x位置的和，也就是前缀和。区间的和是用前缀和减出来的。

怎么做到的？

树状数组的每一个位置管几个位置，位置x写成二进制的时候管最低那个1对应的那个数的位置。
比如29 = 16 + 8 + 4 + 1，所以它管的就是29一个地方；
28 = 16 + 8 + 4，所以它管的是25~28这四个地方。
当我们需要求一个位置x的前缀和的时候，我们就不断地把x的最低1对应的数减去，就可以把区间缩小了。比如要求29的时候，它就分别是16（管1~16）,24（管17~24）,28（管25~28）和29（管29）这四个结点的和。

树状数组的实现：

1. 树状数组是从1开始的，如果输入数组从0开始，要注意下标的映射。
2. 树状数组只能支持把一个位置+/-多少，因此我们需要一个数组维护现在每个位置是几。

//把位置x加上num
void update(int x, int num) 
{
    int i;
    for(i=x; i<=n; i += lowbit(i))
        c[i] += num;//这些位置的c[i]管x，修改的时候往上更新
}

int getsum(int x)
{
    int i;
    int ret=0;
    for(i=x; i>0; i -= lowbit(i)) //不断去掉位置i二进制表示最后的1，例如：13 -> 12 -> 8
        ret += c[i];
    return ret; 
}

void init(vector<int> a)
{
    //初始化树状数组为0，然后以修改的方式初始化树状数组
    int i;
    for(i=0; i<a.size(); i++)
        update(i, a);
}

二维树状数组

和一维完全一样，c[i][j]记录的是矩阵的前缀和，即从左上角(1, 1)到右下角(i, j)的总和。更新和求和的时候，每一维分别向上/向下扩展lowbit那么多即可。

//这里的原始row和col是从0开始的
//这里的val是直接赋值，所以要计算delta
void update(int row, int col, int val) 
{
    int delta = val - num[row][col];
    num[row][col] = val;

    int r,c;
    for(r = row+1; r<=n; r += lowbit(r))
        for(c = col+1; c<=m; c += lowbit(c))
            tree[r][c] += delta;

    return;
}

int getsum(int row, int col)
{
    if(row < 0 || col < 0)
        return 0;

    int r, c;
    int ret = 0;
    for(r=row+1; r>0; r -= lowbit(r))
    {
        for(c = col+1; c>0; c -= lowbit(c))
        {
            ret += tree[r][c];
        }
    }
    return ret;
}

int sumRegion(int row1, int col1, int row2, int col2) 
{       
    return getsum(row2, col2) - getsum(row2, col1-1) - getsum(row1-1, col2) + getsum(row1-1, col1-1);
}