[Vijos] SuperBrother打鼹鼠

背景

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

描述

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

格式

输入格式

每个输入文件有多行。

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

以后每一行开头都有一个数m，表示不同的操作：
m=1，那么后面跟着3个数x,y,k(0<=x,y<n)，表示在点(x,y)处新出现了k只鼹鼠；
m=2，那么后面跟着4个数x1,y1,x2,y2(0<=x1<=x2<n,0<=y1<=y2<n)，表示询问矩形(x1,y1)-(x2,y2)内的鼹鼠数量；
m=3，表示老师来了，不能玩了。保证这个数会在输入的最后一行。

询问数不会超过10000，鼹鼠数不会超过maxlongint。

输出格式

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

样例1

样例输入1

4

1 2 2 5

2 0 0 2 3

3

样例输出1

5

限制

各个测试点1s

提示

水题一道。

所有数据均为随机生成，包括样例……

来源

gnaggnoyil

思路

线段树套线段树维护二维区间和即可；

代码实现

 #include<cstdio>

 #define LL long long

 const int maxm=4e3+;

 inline LL min_(LL x,LL y){return x<y?x:y;}

 inline LL max_(LL x,LL y){return x>y?x:y;}

 int n,m;

 LL t[maxm][maxm];

 void y_add(int k,int l,int r,int x,int y,int p){

     if(l==r){

         t[y][k]+=p;

         return;

     }

     int mid=l+r>>,ls=k<<,rs=ls|;

     if(x<=mid) y_add(ls,l,mid,x,y,p);

     else y_add(rs,mid+,r,x,y,p);

     t[y][k]=t[y][ls]+t[y][rs];

 }

 void x_add(int k,int l,int r,int x,int y,int p){

     if(l==r){

         y_add(,,n,y,k,p);

         return;

     }

     int mid=l+r>>,ls=k<<,rs=ls|;

     if(x<=mid) x_add(ls,l,mid,x,y,p);

     else x_add(rs,mid+,r,x,y,p);

     y_add(,,n,y,k,p);

 }

 LL y_tot(int k,int l,int r,int al,int ar,int p){

     if(l==al&&r==ar) return t[p][k];

     LL ret=;

     int mid=l+r>>,ls=k<<,rs=ls|;

     if(al<=mid) ret+=y_tot(ls,l,mid,al,min_(ar,mid),p);

     if(ar>mid) ret+=y_tot(rs,mid+,r,max_(al,mid+),ar,p);

     return ret;

 }

 LL x_tot(int k,int l,int r,int al,int ar,int x,int y){

     if(l==al&&r==ar) return y_tot(,,n,x,y,k);

     LL ret=;

     int mid=l+r>>,ls=k<<,rs=ls|;

     if(al<=mid) ret+=x_tot(ls,l,mid,al,min_(ar,mid),x,y);

     if(ar>mid) ret+=x_tot(rs,mid+,r,max_(al,mid+),ar,x,y);

     return ret;

 }

 int main(){

     scanf("%d",&n);

     int x1,x2,y1,y2,x,y,k;

     while(scanf("%d",&m),m!=){

         if(m==){

             scanf("%d%d%d",&x,&y,&k);

             x_add(,,n,x,y,k);

         }

         else{

             scanf("%d%d%d%d",&x1,&y1,&x2,&y2);

             printf("%lld\n",x_tot(,,n,x1,x2,y1,y2));

         }

     }

     return ;

 }

这随手写出来的一坨，好像是我平生第一棵树套树QUQ

肚子有点饿，手开始冷了的说。。。