树状数组

树状数组

树状数组是求前n项和的一种非常快捷的方法。一般的求前n项和是从都到尾顺次相加。但是这样在n很大的时候，就显得费时了。

线段树的原理：把n个数从左到右，从低到高构成为二叉树（不一定有同一个根节点，可以有多个根节点，形成多个二叉树）。这颗二叉树的构造可以用公式来确定。奇数都作为叶子结点，每个结点的父结点等于这个结点的下标加上2^k（k是这个结点的高度而非深度）k = 此结点的高度= 这个结点下标写为二进制数末尾0的个数=i&(i^(i - 1))(i是结点下标)=i&(-i)。

下面来讲讲树状数组是怎么实现快速求前n项和的。

   每个结点存放c[]数组元素。

通过图我发现每个结点等于子结点与a[i]的和。所以每个结点都加过子结点的a[i]，那我们在每次把a[i]加入到i结点的父亲中，直到根结点。（第i个结点的父亲等于i+2^k）.

代码如下：

有图知前n项和等于a[1] + .....+ a[n] ,同时也等于不大于n的各二叉树的根结点之和。
i-lowbit(i)也即i-2^k就是i的前一个二叉树的根结点。前n项和是要求出n之前的所有二叉树的根结点。
代码如下：

int sumn(int i)
{
    int sum = 0 ;

    while(i > 0)
    {
        sum += a[i] ;

        i = i - lowbit(i) ;
    }

    return sum ;
}

好了，树状数组就此结束。下面是完整的代码：

#include <iostream>

using namespace std;

#define MAXN 50

int c[MAXN] ;

int a[MAXN] ;

int n ;
int lowbit(int i)
{

    return i & (i ^ (i - 1)) ;

}

void sumc(int i)
{

    while(i <= n)
    {
        c[i] += a[i] ;

        i = i + lowbit(i);
    }

}
int sumn(int i)
{
    int sum = 0 ;

    while(i > 0)
    {
        sum += a[i] ;

        i = i - lowbit(i) ;
    }

    return sum ;
}
void init()
{

    for(int i = 0; i < MXAN ; i ++)
        c[i] = 0 ;
}
int main()
{
    init() ;
    for(int i = 1; i <= n; i ++)
    {
        cin >> a[i] ;

        sumc(i);
    }
    return 0 ;
}