二维线段树(二维树状数组)poj2155

以前写过一个二维线段树,现在忘得差不多了,又想了一遍,就是把一维的每一个节点再抽象成一维的线段树。

这个题查询的时候,每一个x维都要进行y维查询,只要经过的是1的都要加上,因为这样经过的肯定都包含要查询的点

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
const int maxn=1010;
int N,M,x1,x2,Y1,y2,ans;
struct IntervalTree
{
    int cnt[maxn<<2][maxn<<2];
    void build()
    {
        memset(cnt,0,sizeof(cnt));
    }
    void updateY(int xo,int o,int l,int r)
    {
        if(Y1<=l&&r<=y2)
        {
            cnt[xo][o]=!cnt[xo][o];
            return;
        }
        int mid=(l+r)>>1;
        if(Y1<=mid)updateY(xo,o<<1,l,mid);
        if(y2>mid)updateY(xo,o<<1|1,mid+1,r);
    }
    void updateX(int o,int l,int r)
    {
        if(x1<=l&&r<=x2)
        {
            updateY(o,1,1,N);
            return;
        }
        int mid=(l+r)>>1;
        if(x1<=mid)updateX(o<<1,l,mid);
        if(x2>mid)updateX(o<<1|1,mid+1,r);
    }
    void queryY(int xo,int o,int l,int r)
    {
        if(cnt[xo][o])ans++;
        if(l==r)return;
        int mid=(l+r)>>1;
        if(Y1<=mid)queryY(xo,o<<1,l,mid);
        else queryY(xo,o<<1|1,mid+1,r);
    }
    void queryX(int o,int l,int r)
    {
        queryY(o,1,1,N);
        if(l==r)return;
        int mid=(l+r)>>1;
        if(x1<=mid)queryX(o<<1,l,mid);
        else queryX(o<<1|1,mid+1,r);
    }
}tree;
int main()
{
    freopen("in.txt","r",stdin);
    int T;
    char op[5];
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&N,&M);
        tree.build();
        while(M--)
        {
            scanf("%s",op);
            if(op[0]=='C')
            {
                scanf("%d%d%d%d",&x1,&Y1,&x2,&y2);
                tree.updateX(1,1,N);
            }
            else
            {
                scanf("%d%d",&x1,&Y1);
                ans=0;
                tree.queryX(1,1,N);
                printf("%d\n",ans%2);
            }
        }
        if(T)printf("\n");
    }
    return 0;
}

也可以用二维树状数组写

#include
#include
#include
#define N 1005
int c[N][N],n;

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

void update(int x,int y,int val){
    int i,j;
    for(i=x;i<=n;i+=lowbit(i))
        for(j=y;j<=n;j+=lowbit(j))
            c[i][j]+=val;
    return ;
}

int getsum(int x,int y){
    int s,i,j;
    s=0;
    for(i=x;i>=1;i-=lowbit(i))
        for(j=y;j>=1;j-=lowbit(j))
            s+=c[i][j];
    return s;
}

int main(){
    int tt,m,x1,x2,y1,y2;
    char ch[4];
    scanf("%d",&tt);
    while(tt--){
        memset(c,0,sizeof(c));
        scanf("%d%d",&n,&m);
        getchar();
        while(m--){
            scanf("%s",ch);
            if(ch[0]=='C'){
                scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
                x2++;
                y2++;
                update(x2,y2,1);
                update(x1,y1,1);
                update(x1,y2,-1);
                update(x2,y1,-1);
            }else
            if(ch[0]=='Q'){
                scanf("%d%d",&x1,&y1);
                printf("%d\n",getsum(x1,y1)&1);
            }
        }
        printf("\n");
    }
    return 0;
}


你可能感兴趣的:(树状数组/线段树)