树状数组

最新推荐文章于 2024-09-18 20:54:10 发布

漂流瓶终结者

最新推荐文章于 2024-09-18 20:54:10 发布

阅读量314

点赞数

分类专栏：数据结构文章标签：树状数组

数据结构专栏收录该内容

46 篇文章 0 订阅

订阅专栏

引言

在解题过程中，我们有时需要维护一个数组的前缀和S[i]=A[i]+A[2]+...+A[i].但是不难发现，如果我们修改了任意一个A[i],S[i],S[i+1]...S[n]都会发生变化。可以说，每次修改A[i]后，调整前缀和S在最坏的情况下会需要O(n)的时间。当n非常大时，程序会运行得非常慢。因此，这里我们引入“树状数组”，它的修改与求和都是O(logn)，效率非常高。

基本思想

根据任意正整数关于2的不重复次幂的唯一分解性质，若一个正整数x的二进制表示为10101，其中等于1的位是0，2，4，则正整数x可以被"二进制分解成2^4+2^2+2^0"。进一步地，区间[1,x]可以分成O(logx)个小区间:

1.长度为2^4的小区间[1,2^4]

2.长度为2^2的小区间[2^4+1,2^4+2^2]

3.长度为2^0的小区间[2^4+2^2+1,2^4+2^2+2^0]

树状数组就是一种基于上述思想的数据结构，其基本用途是维护序列的前缀和。对于区间[1,x]，树状数组将其分为logx个子区间，从而满足快速询问区间和。

基本算法

由上文可知，这些子区间的共同特点是:若区间结尾为R，则区间长度就等于R的“二进制分解”下最小的2次幂，我们设为lowbit(R)。

对于给定的序列A，我们建立一个数组c，其中c[x]保存序列A的区间[x-lowbit(x)+1,x]中所有数的和。

事实上，数组c可以看做一个如下图所示的树形结构。

该结构满足以下性质：

1.每个内部结点c[x]保存以它为根的子数中所有叶节点的和

2.每个内部结点c[x]的子节点个数等于lowbit(x)的大小

3.除树根外，每个内部结点c[x]的父结点c[x+lowbit(x)]

4.数的深度为O(logN)

求lowbit(x)

lowbit(n)表示取出非负整数n在二进制表示下最低位的1以及它后边的0构成的数值。

lowbit(n)=n&(-n)，这是怎么得来的？具体分析如下:

设n>0，n的第K位是1，第0~k-1位都是0.

为了实现lowbit运算，先把n取反，此时第k位变为0，第0~k-1位都是1.再令n=n+1,此时因为进位，第k位变为1，第0~k-1位都是0.在上面的取反操作后，n的第k+1到最高位恰好与原来相反，所以n&(~n+1)仅有第k位为1，其余位为0.而在补码表示下，~n=-1-n，因此lowbit(n)=n&(-n).

代码实现

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

对某个元素进行加法操作

树状数组支持单点增加，意思是给序列中的某个数A[x]加上y，同时正确维护序列的前缀和。

根据上面给出的树形结构和它的性质，只有结点c[x]及其所有祖先结点保存的"区间和"包含A[x]，而任意一个结点的祖先至多只有logN个，我们逐一对它们的数值c值进行更新即可。下面的代码在O(logN)时间内执行单点增加操作。

代码实现

void update(int x,int y)
{
   for(;x<=N;x+=x&-x)
      c[x]+=y;
}

查询前缀和

树状数组支持查询前缀和，即序列A第1~x个数的和。按照我们刚才提出方法，应该求出x的二进制表示中每个等于1的位。把[1,x]分成O(logN)个小区间，而每个小区间的区间和都已经保存在数组c中。下面的代码是在O(logN)时间内查询前缀和:

int sum(int x)
{
   int ans=0;
   for(;x;x-=x&-x)
      ans+=c[x];
   return ans;
}

统计A[x]....A[y]的值

调用以上的sum操作:sum(y)-sum(x-1)

扩展(多维数组)

一维的树状数组的每个操作的复杂度都是O(logn)的，非常高效。它可以扩充为m维这样每个操作的复杂度就变成了O(log^mn),在m不大的时候，时间复杂度是可以接受的。扩充的方法就是将原来的修改和查询函数中的一个循环，改成m个循环m维数组c中的操作。也就是说，如果有n*m的二维数组a，树状数组为c，那么单点修改操作为:

int update(int x,int y,int z)
{
    int i=x;
    while(i<=n)
    {
        int j=y;
        while(j<=m)
        {
            c[i][j]+=z;
            j+=lowbit(j);
        }
        i+=lobit(i);
    }
}
int sum(int x,int y)
{
    int re=0,i=x;
    while(i>0)
    {
        int j=y;
        while(j>0)
        {
            re+=c[i][j];
            j-=lowbit(j);
        }
        i-=lowbit(i);
    }
    return re;
}