树状数组是一个查询和修改复杂度都为log(n)级别的区间统计的数据结构，在思想上类似于线段树.docVIP

树状数组是一个查询和修改复杂度都为log(n)级别的区间统计的数据结构，在思想上类似于线段树。相比线段树，树状数组需要的空间较少，编程复杂度也较低，但适用范围比线段树小。 来观察一下这个图： 令这棵树的结点编号为C1，C2...Cn。令每个结点的值为这棵树的值的总和，那么容易发现：C1 = A1C2 = A1 + A2C3 = A3C4 = A1 + A2 + A3 + A4C5 = A5C6 = A5 + A6C7 = A7C8 = A1 + A2 + A3 + A4 + A5 + A6 + A7 + A8...C16 = A1 + A2 + A3 + A4 + A5 + A6 + A7 + A8 + A9 + A10 + A11 + A12 + A13 + A14 + A15 + A16这里有一个有趣的性质，下午推了一下发现：设节点编号为x，那么这个节点管辖的区间为2^k（其中k为x二进制末尾0的个数）个元素。因为这个区间最后一个元素必然为Ax，所以很明显：Cn = A(n – 2^k + 1) + ... + An算这个2^k有一个快捷的办法，定义一个函数如下即可： int lowbit(int x) { ???return x(x^(x–1)); } 利用机器补码的特点，这个函数可以改得更方便 int lowbit(int i) { ???return i(-i); } 如果要把a[n]增加m，可以通过调用如下函数实现 void add(int i,int v) { ???while (i=n) ???{ ??????a[i]+=v; ??????i+=lowbit(i); ???} } 如果要统计a[1]到a[n]之间的和，可以通过调用如下函数实现 int sum(int i) { ???int s=0; ???while (i0) ???{ ??????s+=a[i]; ??????i-=lowbit(i); ???} ???return s; } 这是一维的情况，很容易能推广到二维。 POJ上用到树状数组的题目： HYPERLINK /JudgeOnline/problem?id=2352 \t _blank poj2352 Stars HYPERLINK /JudgeOnline/problem?id=2481 \t _blank poj2481 Cows HYPERLINK /JudgeOnline/problem?id=1195 \t _blank poj1195 Mobile phones  HYPERLINK /JudgeOnline/problem?id=3067 \t _blank poj3067 Japan HYPERLINK /JudgeOnline/problem?id=2155 \t _blank poj2155 Matrix HYPERLINK /JudgeOnline/problem?id=2464 \t _blank poj2464 Brownie Points II (以下是经典) 树状数组是一种非常优雅的数据结构. ? 当要频繁的对数组元素进行修改,同时又要频繁的查询数组内任一区间元素之和的时候,可以考虑使用树状数组. ? 最直接的算法可以在O(1)时间内完成一次修改,但是需要O(n)时间来进行一次查询. 而树状数组的修改和查询均可在O(log(n))的时间内完成. ? 假设a[1...N]为原数组,定义c[1...N]为对应的树状数组: c[i] = a[i - 2^k + 1] + a[i - 2^k + 2] + ... + a[i] 其中k为i的二进制表示末尾0的个数,所以2^k即为i的二进制表示的最后一个1的权值. 所以2^k可以表示为n(n^(n-1))或更简单的n(-n): ? int lowbit(int n) {return n (-n);} ? 对a[n]进行修改后,需要相应的修改c数组中的p1, p2, p3...等一系列元素 其中p1 = n, ?????? pi+1 = pi + lowbit(pi) 所以修改原数组中的第n个元素可以实现为: void Modify(int n, int delta) { ??? while(n = N) ??? { c[n] += delta; n += lowbit(n);} } ? 当要查询a[1],a[2]...a[n]的元素之和时,需要累加c数组中的q1, q2, q3...等一系列元素 其中q1? = n, ?????? qi+1 = qi?- lowbit(qi) 所以计算a[1] + a[2] + .. a[n]可以实现为: int Sum(int n) {