HDU2642 Stars 二维树状数组

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2642

题目大意：Yifenfei是一个浪漫的人，他喜欢数天上的星星。为了使问题变得更容易，我们假设天空是一个二维平面，上面的星星有时会亮，有时会发暗。最开始，没有明亮的星星在天空中，然后将给出一些信息，“B XY”，其中“B”代表明亮，x代表X坐标和Y代表Y坐标是指在（X，Y）的明星是光明的，而在“D XY”'D'的意思是灰暗的星星在（X , Y）.当得到“Q X1 X2 Y1 Y2”的查询，你应该告诉Yifenfei在该地区有多少明亮的星星在X1，X2，Y1，Y2决定的矩形里。

实现代码如下：

/*
二维树状数组
B x y：点(x,y)处的星星变亮
D x y：点(x,y)处的星星变暗
Q x1 x2 y1 y2：查询左下角(x1,y1)和右上角(x2,y2)的矩形
*/
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int M=1010;
int a[M][M];
bool vis[M][M];
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 t,x,y;
    char op[5];
    memset(vis,0,sizeof(vis));
    scanf("%d",&t);
    while(t--)
    {
        scanf("%s",op);
        if(op[0]=='B')
        {
            scanf("%d%d",&x,&y);
            x++; y++;
            if(!vis[x][y])
            {
                vis[x][y]=true;
                update(x,y,1);
            }
        }
        else if(op[0]=='D')
        {
            scanf("%d%d",&x,&y);
            x++; y++;
            if(vis[x][y])
            {
                vis[x][y]=false;
                update(x,y,-1);
            }
        }
        else
        {
            int x1,y1;
            scanf("%d%d%d%d",&x,&x1,&y,&y1);
            x++; y++; x1++; y1++;
            if(x<x1) swap(x,x1);
            if(y<y1) swap(y,y1);
            printf("%d\n",query(x,y)-query(x,y1-1)-query(x1-1,y)+query(x1-1,y1-1));
        }
    }
    return 0;
}