POJ2155 Matrix 【二维树状数组】+【段更新点查询】

Description

Input

Output

Sample Input

1
2 10
C 2 1 2 2
Q 2 2
C 2 1 2 1
Q 1 1
C 1 1 2 1
C 1 2 1 2
C 1 1 2 2
Q 1 1
C 1 1 2 1
Q 2 1

Sample Output

1
0
0
1

题意：对一个给定size且初始化为0的矩阵。运行一些命令，Q A B为查看arr[a][b]元素的值，C X1 Y1 X2 Y2为将(x1, y1) (x2, y2)矩形范围内的全部点0、1翻转。

题解：树状数组模式二的使用方法。段更新，点查询。update(x2, y2)表示从(1, 1)到(x2, y2)范围内的全部点都要翻转一次，可是这样会把给定范围外的一些点也翻转到，因此须要将这些点翻转回去。

#include <stdio.h>
#include <string.h>
#define maxn 1002

int size, tree[maxn][maxn];

int lowBit(int x){ return x & (-x); }

//向下更新表示A[1]...A[i]每一个元素都要 += val,推广到二维同理
void update(int x, int y, int val)
{
	int temp;
	while(x > 0){
		temp = y;
		while(temp > 0){
			tree[x][temp] += val;
			temp -= lowBit(temp);
		}
		x -= lowBit(x);
	}
}

int query(int x, int y)
{
	int sum = 0, temp;
	while(x <= size){
		temp = y;
		while(temp <= size){
			sum += tree[x][temp];
			temp += lowBit(temp);
		}
		x += lowBit(x);
	}
	return sum;
}

int main()
{
	//freopen("stdin.txt", "r", stdin);
	
	int cas, q, a, b, c, d;
	char com[2];
	scanf("%d", &cas);
	
	while(cas--){
		scanf("%d%d", &size, &q);
		memset(tree, 0, sizeof(tree));
		
		while(q--){
			scanf("%s%d%d", com, &a, &b);
			if(com[0] == 'C'){
				scanf("%d%d", &c, &d);
				update(c, b - 1, -1);
				update(a - 1, d, -1);
				update(a - 1, b - 1, 1);
				update(c, d, 1);
			}else printf("%d\n", query(a, b) & 1);
		}
		
		if(cas) printf("\n");
	}
	return 0;
}