POJ 1195 二维树状数组

二维树状数组的模版题。写这个题的时候注意不要把a数组直接理解为普通的二维矩阵，了因为i和j的累加和累减方式不同，所以实际上不是一格格的跑的。应该按照一维的树状数组的推广来理解即可。

#include<cstdio>
#include<cstring>
#include<algorithm>


using namespace std;

const int maxn=2000;
int num,n,x,y,z,h;
int a[maxn][maxn];

int lowbit(int x)
{
    return x&(-x);
}

void add(int x,int y,int z)
{
    for(int i=x;i<=n;i+=lowbit(i))
    {
        for(int j=y;j<=n;j+=lowbit(j))
        {
            a[i][j]+=z;
        }
    }
}

int sum(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+=a[i][j];
        }
    }
    return ans;

}

int main()
{
    while(scanf("%d",&num)!=EOF)
    {
        if(num==3)
            break;
        else if(num==0)
        {
            scanf("%d",&n);
            memset(a,0,sizeof(a));
        }
        else if(num==1)
        {
            scanf("%d%d%d",&x,&y,&z);
            add(x+1,y+1,z);
        }
        else if(num==2)
        {
           scanf("%d%d%d%d",&x,&y,&z,&h);
           int ans=0;
            ans=sum(z+1,h+1)-sum(z+1,y)-sum(x,h+1)+sum(x,y);
           printf("%d\n",ans);
        }
    }
    return 0;
}