[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;}