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;
}

猜你喜欢

转载自blog.csdn.net/sirius_han/article/details/81121950