题目链接:http://poj.org/problem?id=2155
题目大意:给定一个n*n的01矩阵,初始时所有元素为0;若输入‘C’,输入两个坐标(x1,y1),(x2,y2),分别代表取反区域的左上角和右下角坐标,将取反区域内0值转为1,1值转为0;若输入‘Q’,输出输入坐标指向的矩阵元素的值;
思路:二维树状数组,操作与一维数组类似,注意a[i][j]储存的是取反的次数而不是取反后的值,执行询问操作时将其对2取余即可;取反操作利用update函数实现,分段取反,偶数次取反相当于于未改变,sum函数用于统计a[x][y]以及其管辖范围内元素的取反次数;
代码实现:
#include<stdio.h>
#include<string.h>
int a[1111][1111];
int N;
int lowbit(int x)
{
return x & (-x);
}
void update(int x, int y)//更新所有管辖范围包括a[x][y]的元素取反次数
{
int i, j;
for(i = x; i <= N; i += lowbit(i))
for(j = y; j <= N; j += lowbit(j))
a[i][j] ++;
}
int sum(int x, int y)
{
int i, j;
int sum = 0;
for(i = x; i > 0; i -= lowbit(i))
for(j = y; j > 0; j -= lowbit(j))
sum += a[i][j]; //统计取反次数
return sum;
}
int main()
{
int x, T, ans;
int i, j, x1, y1, x2, y2;
char op[5];
scanf("%d", &x);
while(x--)
{
memset(a, 0, sizeof(a));
scanf("%d %d", &N, &T);
for(i = 1; i <= T; i++)
{
scanf("%s", op);
if(op[0] == 'C')
{
scanf("%d %d %d %d",&x1, &y1, &x2, &y2);
update(x1, y1);
update(x2 + 1, y1);
update(x1, y2 + 1);
update(x2 + 1, y2 + 1); //分段取反,翻转次数为偶相当与没翻,画图理解
}
else if(op[0] == 'Q')
{
scanf("%d %d",&x1, &y1);
ans = sum(x1, y1) & 1; //位运算,相当于对二取余
printf("%d\n",ans);
}
}
if(x)
printf("\n");
}
return 0;
}