POJ1195 Mobile phones 二维树状数组

题目链接：http://poj.org/problem?id=1195

题目大意：给出一个n*n的矩阵，两种操作：“1 x y v”表示将矩阵上的点(x,y)加上v；“2 x1 y1 x y”表示查询矩阵的和。

分析：和HDU2645 Stars类似，很裸的二维树状数组的题。

实现代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int M=1100;
int a[M][M],n;
int lowbit(int i)
{
    return i&(-i);
}
void update(int x,int y,int v)
{
    for(int i=x;i<M;i+=lowbit(i))
      for(int j=y;j<M;j+=lowbit(j))
        a[i][j]+=v;
}
int query(int x,int y)
{
    int sum=0;
    for(int i=x;i>0;i-=lowbit(i))
      for(int j=y;j>0;j-=lowbit(j))
        sum+=a[i][j];
    return sum;
}
int main()
{
    int op,x,y,v,x1,y1;
    while(true)
    {
        scanf("%d",&op);
        switch(op)
        {
            case 0:scanf("%d",&n);
                   memset(a,0,sizeof(a));
                   break;
            case 1:scanf("%d%d%d",&x,&y,&v);
                   update(x+1,y+1,v);
                   break;
            case 2:scanf("%d%d%d%d",&x1,&y1,&x,&y);
                   x++; y++; x1++; y1++;
                   printf("%d\n",query(x,y)-query(x,y1-1)-query(x1-1,y)+query(x1-1,y1-1));
                   break;
            case 3:return 0;

        }
    }
}