作者:鲸住我了王晨
树状数组:用于单点修改,局部查询等等。
如图:
a是原数组,t/c是我们需要算出来的树状数组
首先我们如何算出c数组的各个值:
如以下add函数所示,我们从1~n,把a数组的各元素传入add函数,注意!这里不是调用一次add函数就算出一个c数组的元素,而是调用a数组的元素进去,修改与这个元素相关的c数组中的元素的值,例如我们传a[1]进去(第94行),然后我们看到add函数主体,data=a[1],c1+=data,经过lowbit函数操作后x=1+1=2,然后c[2]+=data,lowbit函数操作后x=2++2=4,所以c[4]+=data,以此类推(这里lowbit函数稍后再讲,反正就是操作之后可以得到我们想要的数的一个函数)。
然后是单点修改:
我们想修改一个a数组元素的值,但修改后c数组的某些元素也需要跟着修改,我们如何操作呢? 102行,这其实和第一个步骤相同,假如对a[4]增加或修改,那么我们再调用一次add函数,因为add函数内部是+=,修改a[4]后,我们只需要修改与a[4]有关的c数组中的元素的值即可。
区间查询:
如sum函数内部代码:假如要查询1-5的a数组元素的和,res从0开始一直添加,res=c[5]+c[4],这个4是我们通过lowbit函数得到的,如上图,前5项和刚好是c[5]+c[4]。
最后我们来讲解一下lowbit函数:
我们可以看到,传进去x传出的是x&-x,这又是为什么呢?
&是按位与符号,两个二进制数按位与,只有同一位数同时为1,结果才为1,否则为0
例如7,二进制位111,-x就是源码的反码+1,也就是000+1--->001,按位与,结果为001,按照sum函数(128行),x=7-1=6,然后我们看看上图,前7项和确实求了c[7]后要求c[6],这是不是偶然呢,我们再往后求,6的二进制数是110,001+1--->010,按位与结果位010,换成十进制是2,6-2=4,看看上图确实是+c[4],然后4二进制位100,001+1--->100也就是十进制的4,4-4=0刚好退出(125行)的循环,是不是很神奇!!!多看看就懂了,我也是这么过来的。反正我认为这是一个很神奇的算法吧
代码实现: