https://www.bnuoj.com/v3/contest_show.php?cid=9148#problem/F

【題意】

給定一個矩陣，每個格子的初始值為1。現在可以對矩陣有四種操作：

A x y n1 :給格點(x,y)的值加n1

D x y n1: 給格點(x,y)的值減n1,如果現在格點的值不夠n1，把格點置0

M x1 y1 x2 y2：(x1,y1)移動給(x2,y2)n1個

S x1 y1 x2 y2 查詢子矩陣的和

【思路】

當然是二維樹狀數組

但是一定要註意：lowbit(0)是死循環，一定不能是0。所以初始化格點為1的時候要從1開始，以及對於輸入的坐標，我們要加1處理。

【Accepted】

  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 }

【知識點】

1. 樹狀數組是O(logn)的，是因為n的二進制裏最多有logn個1

2. 註意：樹狀數組的下標必須從1開始，，因為lowbit(0)=0,如果從0開始的話就會陷入死循環！！樹狀數組適用於所有滿足結合律的運算（加法，乘法，異或等）

3. 所有樹狀數組能完成的操作線段樹都能夠完成，但是線段樹的代碼復雜，時間復雜度也比較高，查詢、修改需要遞歸完成，而，樹狀數組的操作不僅代碼簡潔，便於理解，而且一切都是遞推完成的，所以能用樹狀數組解決的問題盡量不要用線段樹來寫。

4. 樹狀數組可以查找逆序對，對於LIS問題可以查找方案數。

【二維樹狀數組】See you~