背景
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;
}