POJ 2155：Matrix-二维树状数组，矩阵翻转详细题解

题意

给定一个矩阵，初始化为0，现在可以进行两种操作，一种是查询某个点的值是 0 还是 1。另一种是让这个矩阵的一个子矩阵内的值取反。

思路

首先这题虽然看起来像是一个区间修改，单点查询的题，但是可以转化成单点修改，查询区间和。

• 首先考虑一维的情况，我要一段区间取反，假设是 [l, r]。那么我只需要book[l]+1，book[r+1]+1，假设查询 k 的时候，只需要查询前 k 的和 mod 2 的结果即可。
• 然后这种方法可以推广到二维，不过这里要用一下容斥原理。假设修改的子矩阵左上角和右下角分别为 x1 y1 x2 y2，首先book[x1][y1]+1, book[x2][y2]+1，不过这时要 book[x1][y2+1]+1, book[x2+1][y1]+1。

• 这里红色的就是上面需要标记的点，标记后表示以这个点为左下角的子矩阵内的点全部取反一次。这里绿色代表取反一次，紫色是取反了两次，黄色是取反了四次。最后实际进行取反操作的就是要求取反的子矩阵内的点。
• 注意这里的前缀和用二维树状数组维护，进行单点更新，单点查询。

#include<iostream>
#include<iomanip>
#include<queue>
#include<set>
#include<map>
#include<vector>
#include<string.h>
#include<algorithm>
#include<cmath>
using namespace std;

#define ll long long
#define p pair<ll , ll>
#define MAX 1123

ll n, arr[MAX][MAX], q;

void add(ll x, ll y, ll v) {//更新矩形(a,b)-(c,d)的所有内容的值
	for (ll i = x; i <= n; i += i & -i) {
		for (ll j = y; j <= n; j += j & -j) {
			arr[i][j] += v;
		}
	}
}

ll query(ll x, ll y) {
	ll ans = 0;
	for (int i = x; i > 0; i -= i & -i) {
		for (int j = y; j > 0; j -= j & -j) {
			ans += arr[i][j];
		}
	}
	return ans;
}

int main() {
	ll t; cin >> t;
	while (t--) {
		cin >> n >> q;
		memset(arr, 0, sizeof(arr));
		while (q--) {
			char s; cin >> s; ll a, b, c, d;
			if (s == 'C') {
				cin >> a >> b >> c >> d;
				add(a, b, 1); add(c + 1, b, 1);
				add(a, d + 1, 1); add(c + 1, d + 1, 1);
			}
			else if (s == 'Q') {
				cin >> a >> b;
				cout << query(a, b) % 2 << endl;
			}
		}
		if(t) cout << endl;
	}
}