poj 2155 Matrix(二维树状数组)
链接:http://poj.org/problem?id=2155
题意:给出一个n*n的矩阵,初始化为0,给出q个操作,C x1 y1 x2 y2表示将(x1,y1)至(x2,y2)的矩阵的数翻转(0变成1,1变成0),Q x y表示求(x,y)位置的点的值。
分析:这是hdu1556的二维版本。把(1,1)到四个角的矩阵内的数全翻转1次即update(),然后sum(x,y)即可。
代码如下:
1 #include<cstdio> 2 #include<cstring> 3 const int N=1010; 4 int c[N][N],n; 5 int lowbit(int x) 6 { 7 return x&(-x); 8 } 9 void update(int x,int y,int num) 10 { 11 int i,j; 12 for(i=x;i<=n;i+=lowbit(i)) 13 for(j=y;j<=n;j+=lowbit(j)) 14 c[i][j]^=num; 15 } 16 int sum(int x,int y) 17 { 18 int i,j,s=0; 19 for(i=x;i>0;i-=lowbit(i)) 20 for(j=y;j>0;j-=lowbit(j)) 21 s+=c[i][j]; 22 return s%2; 23 } 24 int main() 25 { 26 int t,q,x1,x2,y1,y2,x,y; 27 char s[10]; 28 scanf("%d",&t); 29 while(t--) 30 { 31 scanf("%d%d",&n,&q); 32 memset(c,0,sizeof(c)); 33 while(q--) 34 { 35 scanf("%s",s); 36 if(s[0]=='C') 37 { 38 scanf("%d%d%d%d",&x1,&y1,&x2,&y2); 39 update(x1,y1,1); 40 update(x2+1,y1,1); 41 update(x1,y2+1,1); 42 update(x2+1,y2+1,1); 43 // update(x1-1,y1-1,1); //超时陷进,树状数组的下标不能为0,所以+1变成上面的代码 44 // update(x2,y1-1,1); 45 // update(x1-1,y2,1); 46 // update(x2,y2,1); 47 } 48 else 49 { 50 scanf("%d%d",&x,&y); 51 printf("%d\n",sum(x,y)); 52 } 53 } 54 printf("\n"); 55 } 56 return 0; 57 }
posted on 2013-08-18 00:32 jumpingfrog0 阅读(191) 评论(0) 编辑 收藏 举报