二叉索引树 树状数组

树状数组 维护一个序列 a1 a2 a3……an

支持两种操作：

1. sum(int a,int b) a~b的区间和

2. add(int x,int d) 第x个数增加d

 

设lowbit(x)为x的二进制最右边的1表示的值

如lowbit(38288)=lowbit(1001010110010000)=10000=16

lowbit(i)=i&-i

 

在树状数组中，对于节点i，如果它是左子结点，父节点为i+lowbit(i)；如果它是右子节点，那么父节点是i-lowbit(i)

（不要在意上面的手写）

设置一个数组C，C[i]=A[i-lowbit(i)+1]+A[i-lowbit(i)+2]+...+A[i]

C对应着那个黑块和它左面的白色长块，如C12=A9+A10+A11+A12

 

那么如何算前缀和S呢？

从i向上、左爬，把经过的C加起来，不一定沿树上的边，如S11=C11+C10+C8

如果修改了一个Ai，如何更新C？

从i开始向上、右爬，修改沿途的C即可。如修改了A3，C3->C4->C8

代码：(短小精悍)

int sum(int x){
    int ans=0;
    while(x>0){
        ans+=C[x];
        x-=lowbit(x);
    }
    return ans;
}

void add(int x,int d){
    while(x<=n){
        C[x]+=d;
        x+=lowbit(x);
    }
}

时间复杂度为log(n)

预处理先把A、C清空，进行n次add，n log n

 

树状数组相比线段树的优势：空间复杂度略低，编程复杂度低，容易扩展到多维情况。

劣势：适用范围小，对可以进行的运算也有限制，比如每次要查询的是一个区间的最小值，似乎就没有很好的解决办法。

 

转载于:https://www.cnblogs.com/FuTaimeng/p/5554474.html