SPOJ - MATSUM Matrix Summation---二维树状数组

题目链接：

https://vjudge.net/problem/SPOJ-MATSUM

题目大意：

二维数组，两种操作

SET 将某点设置成x

SUM 求某个区域之和

解题思路：

这里用二维树状数组

SUM可以直接求出来

这里将某点设置成x，和树状数组不同，树状数组是讲某点加上一个值，但是可以另外建一个数组存储当前所有点的数值，如果要将(x, y)设置成d的话，可以先把该点减去a[x][y]再加上d

合并一下就是：add(x, y, d - a[x][y])

 #include<iostream>
 #include<string>
 #include<cstdio>
 #include<cstring>
 using namespace std;
 typedef long long ll;
 const int maxn = ;
 int tree[maxn][maxn];
 int a[maxn][maxn];//记录当前每个位置的值
 int n;
 int lowbit(int x)
 {
     return x & (-x);
 }
 //修改tree[x][y] += d;
 void add(int x, int y, int d)
 {
     for(int i = x; i <= n; i += lowbit(i))
     {
         for(int j = y; j <= n; j += lowbit(j))
         {
             tree[i][j] += d;
         }
     }
 }
 //从(1,1)到(x, y)数字之和
 ll sum(int x, int y)
 {
     ll ans = ;
     for(int i = x; i > ; i -= lowbit(i))//等于0会无限循环
     {
         for(int j = y; j > ; j -= lowbit(j))
         {
             ans += tree[i][j];
         }
     }
     return ans;
 }

 int main()
 {
     int T;
     scanf("%d", &T);
     while(T--)
     {
         scanf("%d", &n);
         memset(tree, , sizeof(tree));
         memset(a, ,sizeof(a));
         char s[];
         int x, y, z, x1, y1, x2, y2;
         while(scanf("%s", s))
         {
             if(s[] == 'E')break;
             if(s[] == 'E')//SET
             {
                 scanf("%d%d%d", &x, &y, &z);
                 x++;
                 y++;//必须从(1,1)开始
                 add(x, y, z - a[x][y]);//利用之前该位置的数值，就可以把该点设置成z
                 a[x][y] = z;
             }
             else if(s[] == 'U')//SUM
             {
                 scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
                 x1++, y1++, x2++, y2++;
                 ll ans = ;
                 ans += sum(x2, y2);
                 ans += sum(x1 - , y1 - );
                 ans -= sum(x1 - , y2);
                 ans -= sum(x2, y1 - );
                 printf("%lld\n", ans);
             }
         }
     }
     return ;
 }