树状数组总结

树状数组是对一个数组改变某个元素和求和比较实用的数据结构。两中操作都是O(logn)。

在解题过程中，我们有时需要维护一个数组的前缀和S[i]=A[1]+A[2]+...+A[i]。但是不难发现，如果我们修改了任意一个A[i],S[i]、S[i+1]...S[n]都会发生变化。

可以说，每次修改A[i]后，调整前缀和S[]在最坏情况下会需要O(n)的时间。当n非常大时，程序会运行得非常缓慢。

因此，这里我们引入“树状数组”，它的修改与求和都是O(logn)的，效率非常高。

【理论】

          为了对树状数组有个形象的认识，我们先看下面这张图。

          如图所示，红色矩形表示的数组C[]就是树状数组。

          这里，C[i]表示A[i-2^k+1]到A[i]的和，而k则是i在二进制时末尾0的个数，或者说是i用2的幂方和表示时的最小指数。

         （ 当然，利用位运算，我们可以直接计算出2^k=i&(i^(i-1)) ）

          同时，我们也不难发现，这个k就是该节点在树中的高度，因而这个树的高度不会超过logn。

          所以,当我们修改A[i]的值时，可以从C[i]往根节点一路上溯，调整这条路上的所有C[]即可，

          这个操作的复杂度在最坏情况下就是树的高度即O(logn)。  

          另外，对于求数列的前n项和，只需找到n以前的所有最大子树，把其根节点的C加起来即可。

          不难发现，这些子树的数目是n在二进制时1的个数，或者说是把n展开成2的幂方和时的项数,因此，求和操作的复杂度也是O(logn)。

          

树状数组中用的C[]，每个点都有一定的管辖范围；

如C[1]=A[1];             1->0001 最后一位1代表的数字为1，所以是从A[1]向前（包括A[1])共一个数字的和

C[2]=A[1]+A[2];        2->0010 最后一位1代表的数字为2，所以是从A[2]向前共两个数字的和

C[3]=A[3];

C[4]=A[1]+A[2]+A[3]+A[4];

等等；

          接着，我们考察这两种操作下标变化的规律：

          首先看修改操作：

          已知下标i，求其父节点的下标。
          我们可以考虑对树从逻辑上转化：

         如图，我们将子树向右对称翻折，虚拟出一些空白结点（图中白色），将原树转化成完全二叉树。

         有图可知，对于节点i，其父节点的下标与翻折出的空白节点下标相同。

         因而父节点下标 p=i+2^k  (2^k是i用2的幂方和展开式中的最小幂，即i为根节点子树的规模)

         即  p = i + i&(i^(i-1)) 。

         接着对于求和操作：

         因为每棵子树覆盖的范围都是2的幂，所以我们要求子树i的前一棵树，只需让i减去2的最小幂即可。

         即  p = i - i&(i^(i-1)) 。

        

         至此，我们已经比较详细的分析了树状数组的复杂度和原理。

         在最后，我们将给出一些树状数组的实现代码，希望读者能够仔细体会其中的细节。

【代码】

  求最小幂2^k:

int Lowbit(int t)  {      return t & ( t ^ ( t - 1 ) );    //or return t & (-t) ; }

             

这个函数主要是用来求的是某个点管辖范围；

如果是 t+=Lowbit(t) 就是得到的该点的父节点的值；比如t=4时；就能得到8；

而  t-=Lowbit(t)  的话，就是得到t这个点的管辖区间的下个区间的管辖点；

比如说，t=7,代入后6；不断循环到0能依次得到 6.。。4.；他们所有的管辖区间正好是1....7；

  

求前n项和：

int Sum(int end)  {      int sum = 0;      while(end > 0)      {          sum += in[end];          end -= Lowbit(end);      }      return sum;  }

这个函数就是求区间和了。。比如Sum（7）的话，就是求A[1]+A[2]+...A[7]；求一个一般区间的和如 sum( 4,9)可以化成 Sum(9)- Sum(3)来做。

 对某个元素进行加法操作： 

void plus(int pos , int num) 
{ 
    while(pos <= n) 
    { 
          in[pos] += num; 
          pos += Lowbit(pos); 
    } 

} 

这个函数，是用来修改树状数组的；

如果是一般的算法只用修改改点就可；但是树状数组必须修改所有改点被管辖的区间；

比如把a数组的 a[2]减去1，（令N=16）；则所有2被管辖的点有4，8，16都应该减去1；

就是调用函数 update（2，-1）；

资料来自http://blog.csdn.net/shahdza/article/details/6314818

和http://www.cppblog.com/Ylemzy/articles/98322.html