POJ1195 - Mobile phones

题目大意

给定一个N*N的网格，刚开始每个网格的值都是0，接下来会对这些网格进行操作，有一下两种操作：

1、”X Y A“对网格C[x][y]增加A

2、”L B R T“ 查询所有(L<=X<=R,B<=Y<=T)的网格C[X[Y]，并返回它们的总和

题解

非常经典的二维树状数组题目，直接看代码吧

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#define MAXN 1050
using namespace std;
int c[MAXN][MAXN];
int n;
int lowbit(int x)
{
    return x&-x;
}
int sum(int x,int y)
{
    int i,j,ret=0;
   for(i=x;i>0;i-=lowbit(i))
   for(j=y;j>0;j-=lowbit(j))
   ret+=c[i][j];
   return ret;
}
void add(int x,int y,int d)
{
    int i,j;
        for(i=x;i<=n;i+=lowbit(i))
        for(j=y;j<=n;j+=lowbit(j))
        c[i][j]+=d;
}
int main(void)
{
    int a,j,lb,x1,y1,x2,y2,ans;
    while(cin>>a>>n)
    {
        memset(c,0,sizeof(c));
        while(scanf("%d",&lb)==1&&lb!=3)
        {
            if(lb==2)
            {
                  ans=0;
                  scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
                  x1++; y1++;
                  x2++; y2++;
                ans+=sum(x2,y2);
                ans-=sum(x1-1,y2);
                ans-=sum(x2,y1-1);
                ans+=sum(x1-1,y1-1);
                printf("%d\n",ans);
            }
            else
            {
                scanf("%d%d%d",&x1,&y1,&j);
                x1++;
                y1++;
                add(x1,y1,j);
            }
        }
    }
    return 0;
}