POJ 1195 Mobile phones(二维树状数组）

题目链接

赤裸裸的二维树状数组，不用变形什么的。二维树状数组p[i][j]代表的i-lowbit(i)+1 to i ,j-lowbit(j)+1 to j 这个矩阵的和，insert和getsum函数做了一下改变，用两个for来实现插入和求和。又忘记 树状数组 不能为 0，调试了一会，再就是 对求某个区间的和，还是不是很熟悉wa了几次。如求（x1,y1)到（x2,y2)的和，画画图像可以很快得到公式。

getsum(x2,y2)+getsum(x1-1,y1-1)-getsum(x2,y1-1)-getsum(x1-1,y2);

#include <stdio.h>
#include <string.h>
int p[1025][1025],m;
int lowbit(int t)
{
    return (-t)&t;
}
void insert(int x,int y,int d)
{
    int i,j;
    for(i = x;i <= m;i += lowbit(i))
    {
        for(j = y;j <= m;j += lowbit(j))
        p[i][j] += d;
    }
}
int getsum(int x,int y)
{
    int sum = 0,i,j;
    for(i = x;i >= 1;i -= lowbit(i))
    {
        for(j = y;j >= 1;j -= lowbit(j))
        sum += p[i][j];
    }
    return sum;
}
int main()
{
    int n,x,y,a,l,b,r,t;
    for(;;)
    {
        scanf("%d",&n);
        if(n == 3)break;
        if(n == 0)
        {
            scanf("%d",&m);
        }
        else if(n == 1)
        {
            scanf("%d%d%d",&x,&y,&a);
            insert(x+1,y+1,a);
        }
        else if(n == 2)
        {
            scanf("%d%d%d%d",&l,&b,&r,&t);
            printf("%d\n",getsum(r+1,t+1)+getsum(l,b)-getsum(l,t+1)-getsum(r+1,b));
        }
    }
    return 0;
}