POJ 1195 Mobile phones【二维树状数组】

<题目链接>

题目大意：

一个由数字构成的大矩阵，开始是全0,能进行两种操作
1) 对矩阵里的某个数加上一个整数（可正可负）
2) 查询某个子矩阵里所有数字的和
要求对每次查询，输出结果

解题分析：

二维树状数组模板题，需要注意的是，由于题目给的x,y坐标可以为0，所以我们应该将这些点的坐标全部+1，然后就是查询指定子矩阵中所有元素之和，很直观的就能得到式子 ans=sum(x2,y2)-sum(x1-1,y2)-sum(x2,y1-1)+sum(x1-1,y1-1)。

 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 using namespace std;

 const int M = 2e3;
 int row,col,tr[M][M];
 int lowbit(int x){return x&(-x);}
 void add(int i,int j,int val){
     int tmpj;
     while(i<=row){
         tmpj=j;
         while(tmpj<=col){
             tr[i][tmpj]+=val;
             tmpj+=lowbit(tmpj);
         }
         i+=lowbit(i);
     }
 }
 int sum(int i,int j){
     int tmpj,ans=;
     while(i>=){
         tmpj=j;
         while(tmpj>=){
             ans+=tr[i][tmpj];
             tmpj-=lowbit(tmpj);
         }
         i-=lowbit(i);
     }
     return ans;
 }
 int main(){
     int op,n;
     while(scanf("%d%d",&op,&n)!=EOF){
         row=col=n;
         memset(tr,,sizeof(tr));    //初始化树状数组
         while(scanf("%d",&op)&&op!=){
             if(op==){
                 int x,y,c;
                 scanf("%d%d%d",&x,&y,&c);
                 x++,y++;
                 add(x,y,c);
             }
             else{
                 int x1,y1,x2,y2;
                 scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
                 x1++,y1++,x2++,y2++;    //记得将这些坐标+1
                 int ans=sum(x2,y2)-sum(x1-,y2)-sum(x2,y1-)+sum(x1-,y1-);   //sum(x,y)代表(1,1)到(x,y)这个子矩阵内所有元素之和
                 printf("%d\n",ans);
             }
         }
     }
     return ;
 }

2018-10-16