简单论一下区间修改,单点查询的树状数组。
我们是考虑维护一个对于原数组的差分数组,然后取答案时用原数组加上查分数组得到。
二维树状数组
参考资料:资料
单点修改
void modify(int i,int j,int k)
{
a[i][j]+=k;
for(int x=i;x<=n;x+=(x&-x))
{
for(int y=i;y<=m;y+=(y&-y))
{
c[x][y]+=k;
}
}
}
区间查询
int query(int i,int j)
{
int res=0;
for(int x=i;x;x-=(x&-x))
{
for(int y=j;y;y-=(y&-y))
{
res+=c[x][y];
}
}
return res;
}
区间修改
如果用了区间修改,区间查询会降级为单点查询
参考资料: 资料
void range_add(int xa,int ya,int xb,int yb,int k)
{
modify(xa,ya,k);
modify(xa,yb+1,-k);
modify(xb+1,ya,-k);
modify(xb+1,yb+1,k);
}