就像我之前讲二维树状数组单点更新、区间查询一样,二维树状数组与一维没有本质差别,多维的可以通过一维的类推出来。
现在我们由一维树状数组的区间更新、单点查询类推一下二维树状数组的区间更新、单点查询。
首先二维树状数组的区间更新应该是一个矩形区域,我们要更新(x1,y1)到(x2,y2)(x1<=x2,y1<=y2)的矩形区域时由一维类推应该这样:
update(x1,y1)-update(x1,y1+1)-update(x2+1,y1)+update(x2+1,y2+1);
就像这个图,我们想对蓝色区间内修改一次,那么如果我们改变第一个蓝色的点会造成紫色区间内的修改,所以我们必须再通过修改黄、绿、红来抵消这些多余的影响。(容斥原理)
现在以poj2155为模板
题意:给了一个N*N的矩阵(1<=N<=1000)。初始时每个格子里都是0,而对一个格子进行操作是将其0变1或1变0。现在不断地对其一些子矩阵进行操作,并且不断询问某个格子是0还是1。
题解:就是二维树状数组区间更新、单点查询的裸题,直接写即可。这里可以用异或可以简化操作。具体看代码注释。
//#include <bits/stdc++.h>
#include <cstdio>
#include <cstdlib>
#include <cstring>
using namespace std;
const int maxn = 1005;
int sz[maxn][maxn];
int lowbit(int x)
{
return x&(-x);
}
void update(int x,int y)
{
for(int i=x;i<maxn;i+=lowbit(i))
for(int j=y;j<maxn;j+=lowbit(j))
sz[i][j] ^= 1; //从0变1,1变0
}
int query(int x,int y)
{
int ans = 0;
for(int i=x;i>0;i-=lowbit(i))
for(int j=y;j>0;j-=lowbit(j))
ans ^= sz[i][j];
return ans; //奇数个1说明就是1,偶数个1为0
}
int main()
{
int T;
int n,q,x1,x2,y1,y2;
char op;
scanf("%d",&T);
while(T--)
{
memset(sz,0,sizeof(sz));
scanf("%d%d",&n,&q);
while(q--)
{
getchar();
scanf("%c",&op);
if(op=='C')
{
scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
update(x1,y1);
update(x1,y2+1);
update(x2+1,y1);
update(x2+1,y2+1);
}
else
{
scanf("%d%d",&x1,&y1);
printf("%d\n",query(x1,y1));
}
}
printf("\n");
}
return 0;
}