看过这题之后，首先想到的是二维线段树，但是后来做的时候发现不太好搞，因为x和y会相互影响的。
搜了下解题报告，说是用二维树状数组，然后我就自己开始搞。
写过之后虽然AC了，但是跑了1000ms+，感觉太慢了，就搜了下别人的代码，发现我们所理解的二维树状数组有些差别。
我是把每一列当成一个一维的树状数组，等于是n多的一维树状数组，然后这样来看成二维的。
而别人的直接就是用二维来进行查询更新的。
直接看代码吧：
“伪二维树状数组”：
View Code
1 # include<stdio.h> 2 # include<string.h> 3 # define N 1005 4 int sum[N][N],count[N][N]; 5 int Max(int a,int b) 6 { 7 return a>b?a:b; 8 } 9 void insert(int i,int j,int num) 10 { 11 int t1; 12 for(t1=j;t1<=1001;t1+=t1&(-t1)) 13 sum[i][t1]+=num; 14 } 15 int query(int i,int j) 16 { 17 int num=0; 18 while(j>=1) 19 { 20 num+=sum[i][j]; 21 j-=j&(-j); 22 } 23 return num; 24 } 25 int main() 26 { 27 int i,j,ncase,t,n; 28 int x1,y1,x2,y2,num; 29 int MAXx,MAXy,MINx,MINy,ans1,ans2,SS; 30 char word[10]; 31 scanf("%d",&ncase); 32 for(t=1;t<=ncase;t++) 33 { 34 scanf("%d",&n); 35 memset(sum,0,sizeof(sum)); 36 for(i=1;i<=1001;i++) 37 for(j=1;j<=1001;j++) 38 { 39 count[i][j]=1; 40 insert(i,j,1); 41 } 42 printf("Case %d:\n",t); 43 while(n--) 44 { 45 scanf("%s",word); 46 if(word[0]=='A') 47 { 48 scanf("%d%d%d",&x1,&y1,&num); 49 x1++; 50 y1++; 51 count[x1][y1]+=num; 52 insert(x1,y1,num); 53 } 54 else if(word[0]=='D') 55 { 56 scanf("%d%d%d",&x1,&y1,&num); 57 x1++; 58 y1++; 59 if(count[x1][y1]<num) 60 { 61 num=count[x1][y1]; 62 count[x1][y1]=0; 63 } 64 else count[x1][y1]-=num; 65 insert(x1,y1,-num); 66 } 67 else if(word[0]=='M') 68 { 69 scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&num); 70 x1++; 71 y1++; 72 x2++; 73 y2++; 74 if(count[x1][y1]<num) 75 { 76 num=count[x1][y1]; 77 count[x1][y1]=0; 78 } 79 else count[x1][y1]-=num; 80 count[x2][y2]+=num; 81 insert(x1,y1,-num); 82 insert(x2,y2,num); 83 } 84 else 85 { 86 scanf("%d%d%d%d",&x1,&y1,&x2,&y2); 87 x1++; 88 y1++; 89 x2++; 90 y2++; 91 MAXx=Max(x1,x2); 92 MINx=x1+x2-MAXx; 93 MAXy=Max(y1,y2); 94 MINy=y1+y2-MAXy; 95 SS=0; 96 for(i=MINx;i<=MAXx;i++) 97 { 98 ans1=query(i,MAXy); 99 ans2=query(i,MINy-1); 100 SS+=ans1-ans2; 101 } 102 printf("%d\n",SS); 103 } 104 } 105 } 106 return 0; 107 }
二维树状数组：
View Code
View Code 1 # include<stdio.h> 2 # include<string.h> 3 # define N 1005 4 int sum[N][N],count[N][N]; 5 int Max(int a,int b) 6 { 7 return a>b?a:b; 8 } 9 void insert(int i,int j,int num) 10 { 11 int t1,t2; 12 for(t1=i;t1<=1001;t1+=t1&(-t1)) 13 for(t2=j;t2<=1001;t2+=t2&(-t2)) 14 sum[t1][t2]+=num; 15 } 16 int query(int i,int j) 17 { 18 int num=0,t1,t2; 19 for(t1=i;t1>0;t1-=t1&(-t1)) 20 for(t2=j;t2>0;t2-=t2&(-t2)) 21 num+=sum[t1][t2]; 22 return num; 23 } 24 int main() 25 { 26 int i,j,ncase,t,n; 27 int x1,y1,x2,y2,num; 28 int MAXx,MAXy,MINx,MINy,ans1,ans2,SS; 29 char word[10]; 30 scanf("%d",&ncase); 31 for(t=1;t<=ncase;t++) 32 { 33 scanf("%d",&n); 34 memset(sum,0,sizeof(sum)); 35 for(j=1;j<=1001;j++) 36 for(i=1;i<=1001;i++) 37 { 38 count[i][j]=1; 39 sum[i][j]=(i&(-i)) * (j&(-j)); 40 } 41 printf("Case %d:\n",t); 42 while(n--) 43 { 44 scanf("%s",word); 45 if(word[0]=='A') 46 { 47 scanf("%d%d%d",&x1,&y1,&num); 48 x1++; 49 y1++; 50 count[x1][y1]+=num; 51 insert(x1,y1,num); 52 } 53 else if(word[0]=='D') 54 { 55 scanf("%d%d%d",&x1,&y1,&num); 56 x1++; 57 y1++; 58 if(count[x1][y1]<num) 59 { 60 num=count[x1][y1]; 61 count[x1][y1]=0; 62 } 63 else count[x1][y1]-=num; 64 insert(x1,y1,-num); 65 } 66 else if(word[0]=='M') 67 { 68 scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&num); 69 x1++; 70 y1++; 71 x2++; 72 y2++; 73 if(count[x1][y1]<num) 74 { 75 num=count[x1][y1]; 76 count[x1][y1]=0; 77 } 78 else count[x1][y1]-=num; 79 count[x2][y2]+=num; 80 insert(x1,y1,-num); 81 insert(x2,y2,num); 82 } 83 else 84 { 85 scanf("%d%d%d%d",&x1,&y1,&x2,&y2); 86 x1++; 87 y1++; 88 x2++; 89 y2++; 90 MAXx=Max(x1,x2); 91 MINx=x1+x2-MAXx; 92 MAXy=Max(y1,y2); 93 MINy=y1+y2-MAXy; 94 SS=0; 95 SS=query(MAXx,MAXy)-query(MAXx,MINy-1)-query(MINx-1,MAXy)+query(MINx-1,MINy-1); 96 printf("%d\n",SS); 97 } 98 } 99 } 100 return 0; 101 }
