[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

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