spoj 1029 Matrix Summation

题意：

对一个矩阵有2种操作：

1.把某个元素设为x。

2.查询以(x1,y1)为左上角 以(x2,y2)为右上角的矩阵中的数字的和。

思路：

二维树状数组入门题，同时对横坐标和纵坐标做前缀和就行了。

代码：

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
const int N = 1030;
int a[N][N];
int c[N][N];
int n;
int lowbit(int x)
{
    return x&(-x);
}
void add(int x,int y,int ch)
{
    for (int i = x;i <= n;i += lowbit(i))
    {
        for (int j = y;j <= n;j += lowbit(j))
        {
            c[i][j] += ch;
        }
    }
}
int getsum(int x,int y)
{
    int ans = 0;
    for (int i = x;i > 0;i -= lowbit(i))
    {
        for (int j = y;j > 0;j -= lowbit(j))
        {
            ans += c[i][j];
        }
    }
    return ans;
}
int main()
{
    int T;
    scanf("%d",&T);
    while (T--)
    {
        memset(c,0,sizeof(c));
        memset(a,0,sizeof(a));
        scanf("%d",&n);
        char s[5];
        while (scanf("%s",s) != EOF)
        {
            if (s[1] == 'N') break;
            if (s[1] == 'E')
            {
                int x,y,num;
                scanf("%d%d%d",&x,&y,&num);
                x++,y++;
                int ch = num - a[x][y];
                a[x][y] = num;
                add(x,y,ch);
            }
            if (s[1] == 'U')
            {
                int x,y,p,q;
                scanf("%d%d%d%d",&x,&y,&p,&q);
                x++,y++,p++,q++;
                int ans = 0;
                ans += getsum(p,q);
                ans -= getsum(x-1,q);
                ans -= getsum(p,y-1);
                ans += getsum(x-1,y-1);
                printf("%d
",ans);
            }
        }
    }
    return 0;
}