二维树状数组（水题） POJ1195

#include <cstdio>
#include <cstring>

int l,r,x,y,n,a,p,sum[1125][1125];

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

void update(int x, int y, int p)
{
    for(int i=x; i<=n; i+=lowbit(i))
        for(int j=y; j<=n; j+=lowbit(j))
        sum[i][j]+=p;
}

int query(int x, int y)
{
    int ret=0;
    for(int i=x; i>0; i-=lowbit(i))
        for(int j=y; j>0; j-=lowbit(j))
        ret+=sum[i][j];
    return ret;
}

int main()
{
    while(scanf("%d",&p)!=EOF)
    {
        if(p==3) break;
        switch (p) {
            case 0:
                memset(sum,0,sizeof(sum));
                scanf("%d",&n);
                break;
            case 1:
                scanf("%d%d%d",&x,&y,&a);
                update(x+1,y+1,a);
                break;
            case 2:
                scanf("%d%d%d%d",&x,&y,&l,&r);
                printf("%d\n",query(l+1,r+1)-query(l+1,y)-query(x,r+1)+query(x,y));
                break;
        }
    }
    return 0;
}