[POJ 2155] Matrix （二维树状数组）

链接

POJ 2155

---

题意

给出一个N*N的01矩阵，起初均为0，T组询问，一种给出矩形的左上角点和右下角点的坐标，使得矩形内元素取反，另一种给出一个点的坐标，问该点是0还是1。

---

题解

二维树状数组的一道经典题目。
二维树状数组是树状数组在二维的拓展，支持的操作是单点增减值和查询原点(1, 1)到目标点(x, y)的值总和。
但是这道题目是区间增减值和单点查询值，区间值的翻转可以在区间4个端点打一个标记，增1即可，然后查询原点到目标点的值总和，如果包含奇数个1，代表该点翻转了奇数次，应为1，偶数反之。

---

代码

#include <cstdio>
#include <iostream>
using namespace std;
#define maxn (1010)
int c[maxn][maxn];
inline int lowbit(int x) { return x&-x; }
void add(int x, int y, int value, int n)
{
    for(int i = x; i <= n; i += lowbit(i))
    {
        for(int j = y; j <= n; j += lowbit(j))
            c[i][j] += value;
    }
}
int get(int x, int y)
{
    int sum = 0;
    for(int i = x; i; i -= lowbit(i))
    {
        for(int j = y; j; j -= lowbit(j))
            sum += c[i][j];
    }
    return sum;
}
int main()
{
    int X;
    cin >> X;
    while(X--)
    {
        int N, T;
        cin >> N >> T;
        for(int i = 1; i <= N + 1; i++)
        for(int j = 1; j <= N + 1; j++)
            c[i][j] = 0;
        char op[5];
        int x1, x2, y1, y2;
        while(T--)
        {
            scanf("%s", op);
            if(op[0] == 'C')
            {
                scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
                add(x1, y1, 1, N + 1);
                add(x1, y2 + 1, 1, N + 1);
                add(x2 + 1, y1, 1, N + 1);
                add(x2 + 1, y2 + 1, 1, N + 1);
            }
            else
            {
                scanf("%d%d", &x1, &y1);
                printf("%d\n", get(x1, y1) % 2);
            }
        }
        if(X) putchar('\n');
    }
    return 0;
}