Mobile phones
题意:把一个矩阵分为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;
}