树状数组能解决什么问题?
树状数组能高效地维护前缀和,尤其是“查区间改点”类型的前缀和。
如果树状数组是c,原数组是a,那么c[i]表示的是a[i],a[i - lowbit(i)]…这些数的和。
树状数组能直接求的是1~x位置的和,也就是前缀和。区间的和是用前缀和减出来的。
怎么做到的?
树状数组的每一个位置管几个位置,位置x写成二进制的时候管最低那个1对应的那个数的位置。
比如29 = 16 + 8 + 4 + 1,所以它管的就是29一个地方;
28 = 16 + 8 + 4,所以它管的是25~28这四个地方。
当我们需要求一个位置x的前缀和的时候,我们就不断地把x的最低1对应的数减去,就可以把区间缩小了。比如要求29的时候,它就分别是16(管1~16),24(管17~24),28(管25~28)和29(管29)这四个结点的和。
树状数组的实现:
- 树状数组是从1开始的,如果输入数组从0开始,要注意下标的映射。
- 树状数组只能支持把一个位置+/-多少,因此我们需要一个数组维护现在每个位置是几。
//把位置x加上num
void update(int x, int num)
{
int i;
for(i=x; i<=n; i += lowbit(i))
c[i] += num;//这些位置的c[i]管x,修改的时候往上更新
}
int getsum(int x)
{
int i;
int ret=0;
for(i=x; i>0; i -= lowbit(i)) //不断去掉位置i二进制表示最后的1,例如:13 -> 12 -> 8
ret += c[i];
return ret;
}
void init(vector<int> a)
{
//初始化树状数组为0,然后以修改的方式初始化树状数组
int i;
for(i=0; i<a.size(); i++)
update(i, a);
}
二维树状数组
和一维完全一样,c[i][j]记录的是矩阵的前缀和,即从左上角(1, 1)到右下角(i, j)的总和。更新和求和的时候,每一维分别向上/向下扩展lowbit那么多即可。
//这里的原始row和col是从0开始的
//这里的val是直接赋值,所以要计算delta
void update(int row, int col, int val)
{
int delta = val - num[row][col];
num[row][col] = val;
int r,c;
for(r = row+1; r<=n; r += lowbit(r))
for(c = col+1; c<=m; c += lowbit(c))
tree[r][c] += delta;
return;
}
int getsum(int row, int col)
{
if(row < 0 || col < 0)
return 0;
int r, c;
int ret = 0;
for(r=row+1; r>0; r -= lowbit(r))
{
for(c = col+1; c>0; c -= lowbit(c))
{
ret += tree[r][c];
}
}
return ret;
}
int sumRegion(int row1, int col1, int row2, int col2)
{
return getsum(row2, col2) - getsum(row2, col1-1) - getsum(row1-1, col2) + getsum(row1-1, col1-1);
}