【二维树状数组】See you~

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cstring>
  4 #include<string>
  5 #include<algorithm>
  6 #include<cmath>
  7 
  8 using namespace std;
  9 typedef long long ll;
 10 int n;
 11 const int maxn=1e3+5;
 12 int bit[maxn][maxn];
 13 
 14 int lowbit(int x)
 15 {
 16     return (-x)&x;
 17 }
 18 
 19 void add(int x,int y,int val)
 20 {
 21     while(x<maxn)
 22     {
 23         int temp=y;
 24         while(temp<maxn)
 25         {
 26             bit[x][temp]+=val;
 27             temp+=lowbit(temp);
 28         }
 29         x+=lowbit(x);
 30     }
 31 }
 32 
 33 int sum(int x,int y)
 34 {
 35     int ans=0;
 36     while(x>0)
 37     {
 38         int temp=y;
 39         while(temp>0)
 40         {
 41             ans+=bit[x][temp];
 42             temp-=lowbit(temp);
 43         }
 44         x-=lowbit(x);
 45      } 
 46     return ans;
 47 }
 48 
 49 int solve(int x1,int y1,int x2,int y2)
 50 {
 51     return sum(x2,y2)-sum(x1-1,y2)-sum(x2,y1-1)+sum(x1-1,y1-1);
 52 }
 53 
 54 char op[5];
 55 int main()
 56 {
 57     int T;
 58     scanf("%d",&T);
 59     int cas=0;
 60     while(T--)
 61     {
 62         printf("Case %d:\n",++cas);
 63         memset(bit,0,sizeof(bit));
 64         for(int i=1;i<maxn;i++)
 65         {
 66             for(int k=1;k<maxn;k++)
 67             {
 68                 add(i,k,1);
 69             }
 70         }
 71         scanf("%d",&n);
 72         for(int i=0;i<n;i++)
 73         {
 74         //    getchar();
 75             scanf("%s",op);
 76             if(op[0]==‘S‘)
 77             {
 78                 int x1,y1,x2,y2;
 79                 scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
 80                 x1++;y1++;x2++;y2++;
 81                 if(x1>x2)
 82                 {
 83                     swap(x1,x2);
 84                 }
 85                 if(y1>y2)
 86                 {
 87                     swap(y1,y2);
 88                 }
 89                 printf("%d\n",solve(x1,y1,x2,y2));
 90             }
 91             if(op[0]==‘A‘)
 92             {
 93                 int x,y,n1;
 94                 scanf("%d%d%d",&x,&y,&n1);
 95                 x++;y++;
 96                 add(x,y,n1);
 97             }
 98             if(op[0]==‘D‘)
 99             {
100                 int x,y,n1;
101                 scanf("%d%d%d",&x,&y,&n1);
102                 x++;y++;
103                 if(solve(x,y,x,y)<n1)
104                 {
105                     n1=solve(x,y,x,y);
106                 }
107                 add(x,y,-n1);
108             }
109             if(op[0]==‘M‘)
110             {
111                 int x1,y1,x2,y2,n1;
112                 scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&n1);
113                 x1++;y1++;x2++;y2++;
114                 if(solve(x1,y1,x1,y1)<n1)
115                 {
116                     n1=solve(x1,y1,x1,y1);
117                 }
118                 add(x1,y1,-n1);
119                 add(x2,y2,n1);
120             }
121         }    
122     }
123     return 0;    
124 }