hdu 2642二维树状数组 单点更新区间查询 模板题

二维树状数组 单点更新区间查询 模板

从零开始借鉴http://www.2cto.com/kf/201307/227488.html

#include<stdio.h>
#include<string.h>
#define N 1100
int c[N][N],visit[N][N],n;
int number(int x) {
return x&-x;
}
void insert(int x,int y,int z) {
int i,j;
for(i=x;i<=N;i+=number(i))//到最大的范围N，而不是n.
for(j=y;j<=N;j+=number(j))
c[i][j]+=z;
}
int sum(int x,int y) {
int i,j,su=0;
for(i=x;i>0;i-=number(i))
for(j=y;j>0;j-=number(j))
         su+=c[i][j];
return su;
}
int main() {
int i,j,a,b,x1,x2,y1,y2;
char s[3];
while(scanf("%d",&n)!=EOF) {
memset(visit,0,sizeof(visit));
memset(c,0,sizeof(c));
while(n--) {
scanf("%s",s);
if(s[0]=='B') {
scanf("%d%d",&a,&b);
a++;b++;
if(visit[a][b])
continue;
visit[a][b]=1;
insert(a,b,1);
}
if(s[0]=='D') {
scanf("%d%d",&a,&b);
a++;b++;
if(visit[a][b]==0)
continue;
visit[a][b]=0;
insert(a,b,-1);
}
if(s[0]=='Q') {
scanf("%d%d%d%d",&x1,&x2,&y1,&y2);
x1++;x2++;y1++;y2++;
if(x1>x2) {i=x1;x1=x2;x2=i;}
if(y1>y2) {i=y1;y1=y2;y2=i;}
j=sum(x2,y2)-sum(x2,y1-1)-sum(x1-1,y2)+sum(x1-1,y1-1);//用x2,y2范围内的减去x2,y1-1和x1-1,y2范围内的加上多减了一个x1-1,y1-1。

printf("%d ",j);
}
}
}
return 0;
}