POJ 2155 Matrix 二维树状数组

#include<iostream>
#include<math.h>
#include<string.h>
#include<queue>
#include<algorithm>
#include<cstdio>
#define rep(i,a,n) for(i=a;i<=n;i++)
#define per(i,a,n) for(i=n;i>=a;i--)
#define scd1(a) scanf("%d",&a)
#define scd2(a,b) scanf("%d%d",&a,&b)
#define ll long long
#define mem(a,b) memset(a,b,sizeof a);
using namespace std;

const int maxn=1000;
int T,cnt=0,N,M;
int mt[maxn+5][maxn+5];
char ist;

int lowbit(int num){
    return num&(-num);
}

void add(int x, int y, int d){
    while(x<=N){
        int j=y;
        while(j<=N){
            mt[x][j]+=d;j+=lowbit(j);
        }x+=lowbit(x);
    }
}

int sum(int x, int y){
    int ret=0;
    while(x>0){
        int j=y;
        while(j>0){
            ret+=mt[x][j];j-=lowbit(j);
        }
        x-=lowbit(x);
    }
    return ret;
}

int main(){
    scanf("%d",&T);
    while(T--){
        mem(mt,0);
        scanf("%d%d ",&N,&M);
        int i;
        rep(i,1,M){
            scanf("%c",&ist);
            if(ist=='C'){
                int x1,x2,y1,y2;
                scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
                add(x1,y1,1);
                add(x1,y2+1,-1);
                add(x2+1,y1,-1);
                add(x2+1,y2+1,1);
            }
            else if(ist=='Q'){
                int x,y;
                scd2(x,y);
                int ans=sum(x,y);
                printf("%d
",ans%2);
            }
            getchar();
        }
        if(T)printf("
");
    }
}

题意描述: 给出一个矩阵每个点的初值, 每次操作可以对一个子矩阵内的值进行加减, 每次质询要求给出矩阵中某个点的值.

就是一个裸的二维树状数组. 见代码. 它的对偶题是POJ1195 Mobile Phones, 那个题是每次改一个点的值, 每次质询要求给出一个子矩阵内的数值和. 代码只要改一改就能过. 题解链接: http://www.cnblogs.com/LiXinze/p/7353451.html

当然这种题也可以用线段树来做, 只不过稍微会麻烦一点.