vijos-1512 SuperBrother打鼹鼠(二维树状数组)

背景

SuperBrother在机房里闲着没事干(再对比一下他的NOIP,真是讽刺啊……),于是便无聊地开始玩“打鼹鼠”……

描述

在这个“打鼹鼠”的游戏中，鼹鼠会不时地从洞中钻出来，不过不会从洞口钻进去（鼹鼠真胆大……）。洞口都在一个大小为n(n<=1024)的正方形中。这个正方形在一个平面直角坐标系中，左下角为(0,0)，右上角为(n-1,n-1)。洞口所在的位置都是整点，就是横纵坐标都为整数的点。而SuperBrother也不时地会想知道某一个范围的鼹鼠总数。这就是你的任务。

输入格式

每个输入文件有多行。

第一行，一个数n，表示鼹鼠的范围。

以后每一行开头都有一个数m，表示不同的操作：
m=1，那么后面跟着3个数x,y,k(0<=x,y

输出格式

对于每个m=2，输出一行数，这行数只有一个数，即所询问的区域内鼹鼠的个数。

样例输入1

4
1 2 2 5
2 0 0 2 3
3

样例输出1

5

思路

关于二维树状数组:夜深人静写算法（3）：树状数组
这算是二维树状数组的模板题，下面是我的模板：

• n:矩形的大小
• update(x,y,v) 给坐标(x,y)的元素加上值v
• int sum(int x1,int y1,int x2,int y2)，查询这个范围之内的矩阵中元素的和
• query(x,y)的函数返回值代表从(1,1)–>(x,y)的矩阵的数字之和
• 需要注意如果要查询的坐标中有0出现，请给坐标+1

代码

#include <bits/stdc++.h>
using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
const int N=1030+7;
int n;
int c[N][N];
inline int lowbit(int t)
{
    return t&(-t);
}
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))
            c[i][j]+=v;
}
int query(int x,int y)
{
    int s=0;
    for (int i=x; i>0; i-=lowbit(i))
        for (int j=y; j>0; j-=lowbit(j))
            s+=c[i][j];
    return s;
}
int sum(int x1,int y1,int x2,int y2)
{
    x1--,y1--;
    return query(x2,y2)-query(x2,y1)-query(x1,y2)+query(x1,y1);
}
int main()
{
    int m,x1,y1,x2,y2,k,x,y;
    scanf("%d",&n);
    while(scanf("%d",&m)&&(m!=3))
    {
        if(m==1)
        {
            scanf("%d%d%d",&x,&y,&k);
            x++,y++;
            update(x,y,k);
        }
        else if(m==2)
        {
            scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
            x1++,y1++,x2++,y2++;
            printf("%d\n",sum(x1,y1,x2,y2));
        }
    }
    return 0;
}