一)普通树状数组
树状数组(Binary Indexed Tree(BIT))是一个查询和修改复杂度都为log(n)的数据结构。主要用于查询任意两位之间的所有元素之和,单点修改,单点查询。尽管如此,它的常数很小,虽然线段树的许多操作做不到,它依然有着广泛的应用,在辅以CDQ分治、整体二分时尤为凸显。
这就是树状数组的样子。它的实现:
#define lb(x) (x&-x)
int t[MAXN];
void add(int x,int v)
{
while(x<=n)
{
t[x]+=v;
x+=lb(x);
}
}
int query(int x)
{
int ret=0;
while(x)
{
ret+=t[x];
x-=lb(x);
}
return ret;
}
里面的“lb(x)”本来的作用是找出一个数在二进制下最右端的1的位置的,在这里被用于树状数组的转移。
二)二维树状数组
二维树状数组并不难,就是一维的扩展,也就是两层循环:
#define lb(x) (x&-x)
int t[MAXN][MAXM];
void add(int x,int y,int v)
{
int tmp=y;
while(x<=n)
{
while(y<=m)
{
t[x][y]+=v;
y+=lb(y);
}
x+=lb(x);y=tmp;
}
}
int query(int x,int y)
{
int ret=0,tmp=y;
while(x)
{
while(y)
{
ret+=t[x][y];
y-=lb(y);
}
x-=lb(x);y=tmp;
}
return ret;
}
它对于二维的数据适用。
例题:BZOJ1542
#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<math.h>
#define ll long long
#define inf 0x7f7f7f7f
#define lb(x) (x&-x)
using namespace std;
int n,m,q,t[105][305][305],mp[305][305];
int op,x,y,x2,y2,c;
void add(int x,int y,int c,int v)
{
int tmp=y;
while(x<=n)
{
while(y<=m)
{
t[c][x][y]+=v;
y+=lb(y);
}
x+=lb(x);y=tmp;
}
}
int query(int x,int y,int c)
{
int ret=0,tmp=y;
while(x)
{
while(y)
{
ret+=t[c][x][y];
y-=lb(y);
}
x-=lb(x);y=tmp;
}
return ret;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
scanf("%d",&mp[i][j]);
add(i,j,mp[i][j],1);
}
scanf("%d",&q);
for(int i=1;i<=q;i++)
{
scanf("%d",&op);
if(op==1)
{
scanf("%d%d%d",&x,&y,&c);
add(x,y,mp[x][y],-1);
mp[x][y]=c;
add(x,y,mp[x][y],1);
}
else
{
scanf("%d%d%d%d%d",&x,&x2,&y,&y2,&c);
int ans=query(x2,y2,c)+query(x-1,y-1,c);
ans-=(query(x-1,y2,c)+query(x2,y-1,c));
printf("%d\n",ans);
}
}
return 0;
}
三)普通树状数组的区间更新
树状数组总归有缺陷:单点。无论干什么,它只能单点操作,这使它有了极大的局限性。有没有办法改善呢?当然是有的。简单地来说,就是给这个树状数组加一个树状数组,记录它的增量,以实现区间的操作。
具体实现:
#define lb(x) (x&-x)
int t1[MAXN],t2[MAXN];
void add(int *t,int x,int v)
{
while(x<=n)
{
t[x]+=v;
x+=lb(x);
}
}
int query(int *t,int x)
{
int ret=0;
while(x)
{
ret+=t[x];
x-=lb(x);
}
return ret;
}
void add(int l,int r,ll v)
{
add(t1,l,v),add(t1,r+1,-v);
add(t2,l,v*(l-1)),add(t2,r+1,-v*r);
}
int query(int l,int r)
{
return r*query(t1,r)-query(t2,r)
-(l-1)*query(t1,l-1)+query(t2,l-1);
}
例题:POJ3468
四)二维树状数组的区间更新
原理与上面基本相同:
公式化简下来会出现 5个前缀和,分别为:
原数组基数前缀和,常系数前缀和,仅x系数前缀和,仅y系数前缀和,xy系数前缀和。分别维护即可。