SPOJ

题目链接：

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 = 1025;
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 = 0;
    for(int i = x; i > 0; i -= lowbit(i))//等于0会无限循环
    {
        for(int j = y; j > 0; j -= lowbit(j))
        {
            ans += tree[i][j];
        }
    }
    return ans;
}

int main()
{
    int T;
    scanf("%d", &T);
    while(T--)
    {
        scanf("%d", &n);
        memset(tree, 0, sizeof(tree));
        memset(a, 0,sizeof(a));
        char s[5];
        int x, y, z, x1, y1, x2, y2;
        while(scanf("%s", s))
        {
            if(s[0] == 'E')break;
            if(s[1] == '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[1] == 'U')//SUM
            {
                scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
                x1++, y1++, x2++, y2++;
                ll ans = 0;
                ans += sum(x2, y2);
                ans += sum(x1 - 1, y1 - 1);
                ans -= sum(x1 - 1, y2);
                ans -= sum(x2, y1 - 1);
                printf("%lld
", ans);
            }
        }
    }
    return 0;
}