Mobile phones POJ - 1195 （二维树状数组）

Mobile phones

 题目链接：POJ - 1195 

题意：把一个矩阵分为S*S的方格，初始化为0，每次将(x, y)格子里增加a，或者求l<=x<=r&&b<=y<=t的矩阵内方格中数的和；

这个题用二维的树状数组；

其实和一维一样， 就是多个一维的摞起来构成了二维的；

需要注意的是最后求出来的tr[r][t]-tr[l][b]并不是最终结果，如下图：紫色方块是红色方块减去两个蓝色在加上绿色方块；

#include <cstring>
#include <cstdio>
using namespace std;
int tr[1500][1500], n;
int lowbit(int x){
	return x&(-x);
}
void updata(int x, int y, int a){
	for(int i=x; i<=n; i+=lowbit(i)){
		for(int j=y; j<=n; j+=lowbit(j)){
			tr[i][j]+=a;
		}
	}
}
int getsum(int x, int y){
	int s=0;
	for(int i=x; i>0; i-=lowbit(i)){
		for(int j=y; j>0; j-=lowbit(j)){
			s+=tr[i][j];
		}
	}
	return s;
}
int main(){
	int op, x, y, a, l, b, r, t;
	memset(tr, 0, sizeof(tr));
	scanf("%d%d", &op, &n);
	while(scanf("%d", &op)){
		if(op==3) break;
		if(op==1){
			scanf("%d%d%d", &x, &y, &a);
			updata(x+1, y+1, a);
		}
		if(op==2){
			scanf("%d%d%d%d", &l, &b, &r, &t);
			printf("%d\n", getsum(r+1, t+1)-getsum(l, t+1)-getsum(r+1, b)+getsum(l, b));
		}
	}
	return 0;
}