[POJ 2155] Matrix (高维树状数组)

POJ - 2155

给定一个矩阵,开始时其元素全为 0,有两个操作
C x1 y1 x2 y2 将左上角为 (x1,y1),右下角为 (x2,y2)的子矩阵里的元素取反
Q x y 询问(x,y)元素的值

裸的二维树状数组,知道了套路以后还是很好做的
add2(x1,y1,1);add2(x1,y2+1,-1);
add2(x2+1,y1,-1);add2(x2+1,y2+1,1)
根据树状数组在一个地方加值,其后的元素全部都加同一个值
用以上四个操作,把多加的扣掉,多扣的加回来即可

#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <map>
#include <set>
#include <queue>
using namespace std;
typedef pair<int,int> Pii;
typedef long long LL;
typedef unsigned long long ULL;
typedef double DBL;
typedef long double LDBL;
#define MST(a,b) memset(a,b,sizeof(a))
#define CLR(a) MST(a,0)
#define Pow2(a) (a*a)

const int maxn=1e3+10;
int N,M;
int Bit2[maxn][maxn];
int lowbit(int x){return x&-x;}
void add2(int,int,int);
int sum2(int,int);

int main()
{
    int T;
    scanf("%d", &T);
    for(int ck=1; ck<=T; ck++)
    {
        CLR(Bit2);
        scanf("%d%d", &N, &M);
        for(int m=0; m<M; m++)
        {
            char opt;
            int x,y,xx,yy;
            scanf(" %c", &opt);
            if(opt=='C')
            {
                scanf("%d%d%d%d", &x, &y, &xx, &yy);
                add2(x,y,1);add2(xx+1,y,-1);
                add2(x,yy+1,-1);add2(xx+1,yy+1,1);
            }
            else
            {
                scanf("%d%d", &x, &y);
                printf("%d\n", sum2(x,y)&1);
            }
        }
        if(ck<T) puts("");
    }
    return 0;
}

void add2(int x, int y, int v)
{
    for(int i=x; i<=N; i+=lowbit(i))
    {
        for(int j=y; j<=N; j+=lowbit(j))
        {
            Bit2[i][j]+=v;
        }
    }
}

int sum2(int x,int y)
{
    int res=0;
    for(int i=x; i>0; i-=lowbit(i))
    {
        for(int j=y; j>0; j-=lowbit(j))
        {
            res+=Bit2[i][j];
        }
    }
    return res;
}

你可能感兴趣的:(poj)