Light OJ 1266(二维树状数组)

  跟一维的差不多，就是多了一层循环。

需要注意的就是要知道树状数组下标是从0开始的；

这里read函数就是读取一个以(1 , 1)为左下角的顶点的子矩阵的值。

#include<cstdio>
#include<cstring>
using namespace std;
const int N=1e3;
int tree[N+5][N+5];
bool vis[N+5][N+5];
int lowbit(int x)
{
  return (x&(-x));
}
void update(int x,int y,int data)
{
   int i,j;
   for(i=x;i<=N;i+=lowbit(i))
   {
     for(j=y;j<=N;j+=lowbit(j))
       tree[i][j] += data;
   }
}
int read(int x,int y)
{
   int i,j,ans = 0;
   for(i=x;i>0;i-=lowbit(i))
   {
     for(j=y;j>0;j-=lowbit(j))
       ans+=tree[i][j];
   }
   return ans;
}
int main()
{
  //  freopen("in.txt","r",stdin);
    int T,ca=0;
    scanf("%d",&T);
    while(T--)
    {
        int q,ty,x1,x2,y1,y2;
        memset(tree,0,sizeof(tree));
        memset(vis,0,sizeof(vis));
        printf("Case %d:\n",++ca);
        scanf("%d",&q);
        while(q--)
        {
            scanf("%d",&ty);
            if(ty==0)
            {
                scanf("%d%d",&x1,&y1);
                x1++,y1++;
                if(!vis[x1][y1])  vis[x1][y1]=1,update(x1,y1,1);
            }
            else
            {
                scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
                x1++,x2++,y1++,y2++;
                printf("%d\n",read(x2,y2)-read(x1-1,y2)-read(x2,y1-1)+read(x1-1,y1-1));
            }
        }
    }
    return 0;
}