【模板】树状数组

树状数组的问题模型：

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

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

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

 

树状数组的理解：

这是经常看到的一个图，其实可以把他看作是二叉树的形式。

不理解的可以看：点击打开

据图可知：

1=(001)      C[1]=A[1];
2=(010)      C[2]=A[1]+A[2];
3=(011)      C[3]=A[3];
4=(100)      C[4]=A[1]+A[2]+A[3]+A[4];
5=(101)      C[5]=A[5];
6=(110)      C[6]=A[5]+A[6];
7=(111)      C[7]=A[7];
8=(1000)    C[8]=A[1]+A[2]+A[3]+A[4]+A[5]+A[6]+A[7]+A[8];

 

分析上面的几组式子可知，当 i 为奇数时，ci=ai ；当 i 为偶数时，就要看 i 的因子中最多有二的多少次幂，例如，6 的因子中有 2 的一次幂，等于 2 ，所以 c6=a5+a6（由六向前数两个数的和），4 的因子中有 2 的两次幂，等于 4 ，所以 c4=a1+a2+a3+a4（由四向前数四个数的和）。

有公式：cn=a(n-a^k+1)+.........+an（其中 k 为 n 的二进制表示中从右往左数的 0 的个数）。例如i=8时，k=3。

那么，如何求 a^k 呢？

没错，就是用 lowbit() ~.~

这里有两种方法：

方法一：

int lowbit(x) 
{   
    return x - (x & (x - 1));
}

方法二：

int lowbit(x) 
{   
    return x & -x;
}

lowbit（）的返回值就是 2^k 次方的值。重要：lowbit(x) 其实就是取出x的最低位1 。

求出来 2^k 之后，数组 c 的值就都出来了，接下来我们要求数组中所有元素的和。

 

求数组的和的算法如下：

（1）首先，令sum=0，转向第二步；

（2）接下来判断，如果 n>0 的话，就令sum=sum+cn转向第三步，否则的话，终止算法，返回 sum 的值；

（3）n=n - lowbit（n）（将n的二进制表示的最后一个零删掉），回第二步。

代码实现：
 

int Sum(int n)
{
    int sum=0;
    while(n>0)
    {
         sum+=c[n];
         n=n-lowbit(n);
    }    
    return sum;
}

数组更改元素的算法如下：

（1）当 i<=n 时，执行下一步；否则的话，算法结束;

（2）ci=ci+x ，i=i+lowbit（i）（在 i 的二进制表示的最后加零），返回第一步。

代码实现：
 

void change(int i,int x)
{
     while(i<=n)
     {
          c[i]=c[i]+x;
          i=i+lowbit(i);
     }
}

整体代码：

#include<iostream>
using namespace std;
int n,m,i,num[100001],t[200001],l,r;//num:原数组；t：树状数组 
int lowbit(int x)
{
    return x&(-x);
}
void change(int x,int p)//将第x个数加p 
{
    while(x<=n)
    {
        t[x]+=p;
        x+=lowbit(x);
    }
    return;
}
int sum(int k)//前k个数的和 
{
    int ans=0;
    while(k>0)
    {
        ans+=t[k];
        k-=lowbit(k);
    }
    return ans;
}
int ask(int l,int r)//求l-r区间和 
{
    return sum(r)-sum(l-1); 
}
int main()
{
    cin>>n>>m;
    for(i=1;i<=n;i++)
    {
        cin>>num[i];
        change(i,num[i]);
    }
    for(i=1;i<=m;i++)
    {
        cin>>l>>r;
        cout<<ask(l,r)<<endl;
    }
    return 0;
}

树状数组的优点：
1.代码短小，实现简单；
2.容易扩展到高纬度的数据；
缺点：
1.只能用于求和，不能求最大／小值；
2.不能动态插入；
3.数据多时，空间压力大；

 

https://blog.csdn.net/acm_hmj/article/details/53264364

http://www.cnblogs.com/zhangshu/archive/2011/08/16/2141396.html

https://www.cnblogs.com/hsd-/p/6139376.html

https://blog.csdn.net/acm_hmj/article/details/53223678