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，那么后面跟着4个数x1,y1,x2,y2(0<=x1<=x2
m=3，表示老师来了，不能玩了。保证这个数会在输入的最后一行。
询问数不会超过10000，鼹鼠数不会超过maxlongint。

输出格式

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

样例输入

4
1 2 2 5
2 0 0 2 3
3

样例输出

5

题解

二维树状数组模板。

#include
#include
#include
#include
#include
using namespace std;
int n,m,ct[1025][1025];
int lowbit(int x)
{return x&(-x);}
void insert(int x,int y,int z)
{
	for(int i=x;i<=n;i+=lowbit(i))
	for(int j=y;j<=n;j+=lowbit(j))
	   ct[i][j]+=z;
}
int cal(int x,int y)
{
	int ans=0;
	for(int i=x;i>0;i-=lowbit(i))
	for(int j=y;j>0;j-=lowbit(j))
	   ans+=ct[i][j];
	return ans;
}
void init()
{
	scanf("%d",&n);
	while(scanf("%d",&m))
	   {if(m==3) return;
	    else if(m==1)
	       {int x,y,z;
			scanf("%d%d%d",&x,&y,&z); insert(x+1,y+1,z);
		   }
		else
		   {int a,b,c,d;
			scanf("%d%d%d%d",&a,&b,&c,&d);
			printf("%d\n",cal(c+1,d+1)-cal(a,d+1)-cal(c+1,b)+cal(a,b));
		   }
	   }
}
int main()
{init(); return 0;}