树状数组入门

文章转载自：https://blog.csdn.net/flushhip/article/details/79165701

1. 树状数组的问题模型

首先我们搞明白树状数组是用来干嘛的，现在有一个这样的问题：有一个数组a，下标从0到n-1，现在给你w次修改，q次查询，修改的话是修改数组中某一个元素的值；查询的话是查询数组中任意一个区间的和，w + q < 500000。

这个问题很常见，首先分析下朴素做法的时间复杂度，修改是O(1)的时间复杂度，而查询的话是O(n)的复杂度，总体时间复杂度为O(qn)；可能你会想到前缀和来优化这个查询，我们也来分析下，查询的话是O(1)的复杂度，而修改的时候修改一个点，那么在之后的所有前缀和都要更新，所以修改的时间复杂度是O(n)，总体时间复杂度还是O(qn)。

可以发现，两种做法中，要么查询是O(1)，修改是O(n)；要么修改是O(1)，查询是O(n)。那么就有没有一种做法可以综合一下这两种朴素做法，然后整体时间复杂度可以降一个数量级呢？有的，就是树状数组。

2. lowbit函数

这里我们先不管树状数组这种数据结构到底是什么，先来了解下lowbit这个函数，你也先不要问这个函数到底在树状数组中有什么用；

顾名思义，lowbit这个函数的功能就是求某一个数的二进制表示中最低的一位1，举个例子，x = 6，它的二进制为110，那么lowbit(x)就返回2，因为最后一位1表示2。

那么怎么求lowbit呢？

• 第一种方法是先消掉二进制中的最后一位1，然后再用原数减去消掉最后一位1后的数，答案就是lowbit(x)的结果；

• 第二种方法就是求负数的补码的最简便方法：把这个数的二进制写出来，然后从右向左找到第一个1(这个1就是我们要求的结果，但是现在表示不出来，后来的操作就是让这个1能表示出来)，这个1不要动和这个1右边的二进制不变，左边的二进制依次取反，这样就求出的一个数的补码，而不是通过负数的反码加一来得到补码。说这个方法主要是让我们理解一个负数的补码在二进制上的特征。把这个负数补码与对应正数补码(计算机都是转成补码运算，正数补码就是它的原码)进行与运算，由于这个1的左边的二进制与正数原码对应的部分是相反的，所以相与一定都为0,；由于这个1和这个1右边的二进制都是不变的，因此相与后还是原来的样子，相当于截去了从右向左第一个1左边的部分！这样与出来的结果就是lowbit(x)的结果。

两种方法对应的代码依次如下：

//方法一
int lowbit(x) 
{   
    //原数减去消掉二进制最后一位后的数
    return x - (x & (x - 1));
}

//方法二
int lowbit(x) 
{   
    //负数补码与对应正数补码的与运算结果
    return x & -x;
}

3. 树状数组的思想

在树状数组的问题模型中已经有所提及了，就是那两种不同做法的一个综合；

先定义一些东西：arr是原数组，c是新开的一个数组，这个数组代表后缀和(问题模型中是用的前缀和，这里要用后缀和，具体原因马上就知道了)；

二进制的视角：一个数n，假设n = 6，它的二进制为110，我们把它表示成累加的形式110 = 100 + 10，这样是可以的，那么我们要求前6(110)项的和是不是可以这样求：

注意括号中的元素个数，是不是4(100)个加2(10)个，和110 = 100 + 10是不是很像，不知你们发现了吗，10就是lowbit(110)的结果，100是lowbit(100)的结果。求和的时候我们总是把∑ni=1拆分成这样的几段区间和来计算，而如何去确定这些区间的起点和长度呢？就是根据n的二进制来的，二进制怎么拆的，你就怎么拆分，而拆分二进制就要用到上面说的lowbit函数了。这里也可以顺理成章得给出c数组的表示了。

c[i]表示从第i个元素向前lowbit(i)个元素的和，这就是上面说的区间和，只不过这个区间是靠右端点的；那中间的这些区间怎么定义？

你把c[6]去掉，不就是：

这个区间不就靠右端点了吗

4. 树状数组的实现

设计一种数据结构，需要的操作无非就是”增删改查“，这里只讨论查询和修改操作具体是怎么实现的；

查询

这里说的查询是查询任一区间的和，由于区间和具有可加减性，故转化为求前缀和；

查询前缀和刚刚在树状数组的思想中已经说过了，就是把大区间分成几段长度不等的小区间，然后求和。区间的个数为O(logn)，所以查询的时间复杂度为O(logn)。

区间和相减

修改

修改某一位置上的元素的时间复杂度为O(1)，但是要更新c数组，不然查询的时间复杂度就会变高。更新的方法就要提一下树状数组的性质了和树状数组那张经典的图片了。

这张图片中已经把c数组的后缀和这个含义已经表达得很清楚了。这个时候你再把查询操作对应到这张图上，然后看着二进制来操作，是不是就可以很直白地理解上面所说的查询操作了！

我们从这张图中可以得到树状数组的如下性质：

• 后缀和的长度是2的幂；
• 上一层后缀和的长度是下一层后缀和长度的两倍；
• 下一层后缀和只要补上自己后缀和的长度就可以得到上面层的后缀和(图中的虚框框)，注意，是上面的后缀和，而不是上一层的后缀和，这个性质就是更新操作的依据；
• 最后一位1右边有多少个0(可以用log2(lowbit(x))表示)就表示这一层有多少个直系子层(子层的意思就是这一层的和包含下面某一层的和)。

更新的时候只要更新修改这个点会影响到的那些后缀和(c数组)，假设现在修改6(110)这个点，依据树状数组的性质三，它影响的直系父层就是c[6(110) + lowbit(6(110))] = c[8(1000)]，但是它肯定不是只影响直系父层，上面所有包含这一层和的层都要更新，但是我们把这个更新传递给直系父层c[8],8这个点的直系父层是c[16]，依次类推地更新就行了。