【题解】hdu2642（2018-07-25校赛 树状数组-最值-二分-二维G）二维树状数组

题目链接

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1e3+10;
int a[N][N],vis[N][N];
inline int lowbit(int x)
{
    return x&(-x);
}
void update(int x,int y,int v)
{
    for(int i=x;i<N;i+=lowbit(i))
    for(int j=y;j<N;j+=lowbit(j))
    a[i][j]+=v;
}
int query(int x,int y)
{
    int ans=0;
    for(int i=x;i;i-=lowbit(i))
    for(int j=y;j;j-=lowbit(j))
    ans+=a[i][j];
    return ans;
}
int main()
{
    //freopen("in.txt","r",stdin);
    int t,x,y;
    memset(vis,0,sizeof(vis));
    scanf("%d",&t);
    while(t--)
    {
        char op[2];
        scanf("%s",op);
        if(op[0]=='B')
        {
            scanf("%d%d",&x,&y);
            x++;y++;//范围里有1 
            if(!vis[x][y])
            {
                vis[x][y]=1;
                update(x,y,1);
            }
        }
        else if(op[0]=='D')
        {
            scanf("%d%d",&x,&y);
            x++;y++;
            if(vis[x][y])
            {
                vis[x][y]=0;
                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;
}