POJ - 2155 Matrix 二维线段树 或二维树状数组

POJ - 2155

---

目录

一、用线段树求解

二、用二维树状数组求解

---

这道题整了好久orz，弄懂那一刻也是挺满足的。

一维的区间上升到二维，维护一个矩阵。二维线段树其实就就是在第一位区间的每个节点下再建一颗线段树，表示第二维的区间。

如果用线段树的话是不能用lazy标记的，刚开始做的时候不懂，用了下传标记，就是做不出。后来看了很多篇题解慢慢的懂了。。。

 

可以用线段树或树状数组写，树状数组写代码比较简单吧   我是先用线段树写的。 两种数据结构的写法都在下面，线段树我用了两种不同的写法写的。

 

一、用线段树求解

 

二维线段树知识点链接

这道题是要进行 单点查询和矩阵修改操作。

矩阵修改：在修改的时候只需要先找到第一维的对应区间，在在这个区间的弟二维中查找对应区间，再做修改即可。

单点查询：查询的时候把这个点以及和这个点相关的都累加起来。

我们可以在修改时标记某一个节点，那么这个节点以下的区间就都是要修改的，当我们在查找的时候，只需要统计查找到这个点时，一路上有多少个被修改的区间，是偶数说明呗修改回来了，是奇数那就是被修改了。

 

线段树代码1   参考了kuangbin的代码：kuangbin题解链接

#include<iostream>
#include<stdio.h>
#include<cstring>
#include<algorithm>
#include<string>
#include<vector>
using namespace std;
const int maxn=1005;
int n,m;
int ans;
struct nodeY
{
    int l,r;
    int val;
};
struct nodeX
{
    int l,r;
    nodeY sty[maxn<<2];
    void build(int l,int r,int k)
    {
        sty[k].val=0;
        sty[k].l=l;
        sty[k].r=r;
        if(l==r)
            return;
        int mid=(l+r)/2;
        build(l,mid,2*k);
        build(mid+1,r,2*k+1);
    }
    void update(int l,int r,int k,int add)
    {
        if(sty[k].l>=l&&sty[k].r<=r)
        {
            sty[k].val+=add;
            return;
        }
        int mid=(sty[k].l+sty[k].r)/2;

        if(mid>=l) update(l,r,2*k,add);
        if(mid<r) update(l,r,2*k+1,add);
    }
    void query(int l,int r,int k,int y)
    {
        ans+=sty[k].val;
        if(l==r)
            return;
        int mid=(l+r)/2;
        if(mid>=y) query(l,mid,2*k,y);
        else query(mid+1,r,2*k+1,y);
    }
};nodeX stx[maxn<<2];


void build(int l,int r,int k)
{
    stx[k].build(1,n,1);
    stx[k].l=l;
    stx[k].r=r;
    if(l==r)
        return;
    int mid=(l+r)/2;
    build(l,mid,2*k);
    build(mid+1,r,2*k+1);
}
void update(int x1,int x2,int y1,int y2,int k,int add)
{
    if(stx[k].l>=x1&&stx[k].r<=x2)
    {
        stx[k].update(y1,y2,1,add);
        return;
    }
    int mid=(stx[k].l+stx[k].r)/2;
    if(mid>=x1) update(x1,x2,y1,y2,2*k,add);
    if(mid<x2) update(x1,x2,y1,y2,2*k+1,add);
}
void query(int x,int y,int k)
{
    stx[k].query(1,n,1,y);
    if(stx[k].l==stx[k].r)
        return;
    int mid=(stx[k].l+stx[k].r)/2;
    if(x<=mid) query(x,y,2*k);
    else query(x,y,2*k+1);
}
int main()
{
//    freopen("in.txt","r",stdin);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        char op[5];
        int x1,y1,x2,y2;
        int x,y;
        scanf("%d%d",&n,&m);
        build(1,n,1);
        for(int i=0;i<m;i++)
        {
            scanf("%s",op);
            if(op[0]=='C')
            {
                scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
                update(x1,x2,y1,y2,1,1);
            }
            else
            {
                ans=0;
                scanf("%d%d",&x,&y);
                query(x,y,1);
                if(ans%2==1)
                    printf("1\n");
                else
                    printf("0\n");
            }
        }
        if(T) printf("\n");
    }
    return 0;
}

线段树代码2  参考   线段树题解2       二维线段树知识点链接

#include<iostream>
#include<stdio.h>
#include<cstring>
#include<algorithm>
#include<string>
#include<vector>
using namespace std;
const int maxn=1005;
int sum[maxn<<2][maxn<<2];
int n,m;
int x1,y1,x2,y2;
int x,y;
int ans;
void updateY(int l,int r,int kx,int ky,int add)//l,r表示当前区间,二维的所以要写kx和ky
{
    if(l>=y1&&r<=y2)
    {
        sum[kx][ky]+=add;
        return;
    }
    int mid=(l+r)/2;
    if(mid>=y1) updateY(l,mid,kx,2*ky,add);
    if(mid<y2) updateY(mid+1,r,kx,2*ky+1,add);
}
void updateX(int l,int r,int k,int add)//一维的修改
{
    if(l>=x1&&r<=x2)
    {
        updateY(1,n,k,1,add);
        return;
    }
    int mid=(l+r)/2;
    if(mid>=x1) updateX(l,mid,2*k,add);
    if(mid<x2) updateX(mid+1,r,2*k+1,add);
}
void queryY(int l,int r,int kx,int ky)//二维的 所以要写 kx 和ky
{
    ans+=sum[kx][ky];
    if(l==r)
        return;
    int mid=(l+r)/2;
    if(y<=mid) queryY(l,mid,kx,2*ky);
    if(y>mid) queryY(mid+1,r,kx,2*ky+1);
}
void queryX(int l,int r,int k)
{
    queryY(1,n,k,1);
    if(l==r)
        return;
    int mid=(l+r)/2;
    if(x<=mid) queryX(l,mid,2*k);
    if(x>mid) queryX(mid+1,r,2*k+1);
}
int main()
{
    freopen("in.txt","r",stdin);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        memset(sum,0,sizeof(sum));
        char op[5];
        scanf("%d%d",&n,&m);
        for(int i=0;i<m;i++)
        {
            scanf("%s",op);
            if(op[0]=='C')
            {
                scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
                updateX(1,n,1,1);
            }
            else
            {
                ans=0;
                scanf("%d%d",&x,&y);
                queryX(1,n,1);
                if(ans%2==1)
                    printf("1\n");
                else
                    printf("0\n");
            }
        }
        if(T) printf("\n");//注意输出格式
    }
    return 0;
}

 

二、用二维树状数组求解

 

这篇文章讲的树状数组，看完就会写了！！！树状数组链接

#include<iostream>
#include<stdio.h>
#include<cstring>
#include<algorithm>
#include<string>
#include<vector>
using namespace std;
const int maxn=1005;
int n,m;
int x1,y1,x2,y2;
int x,y;
int c[maxn][maxn];
int lowbit(int x){return x&(-x);}
void add(int x,int y,int num)
{
    for(int i=x;i<=n;i+=lowbit(i))
        for(int j=y;j<=n;j+=lowbit(j))
            c[i][j]+=num;
}
void range_add(int x1,int x2,int y1,int y2,int num)
{
    add(x1,y1,num);
    add(x1,y2+1,-num);
    add(x2+1,y1,-num);
    add(x2+1,y2+1,num);
}
int sum(int x,int y)
{
    int ans=0;
    for(int i=x;i>0;i-=lowbit(i))
        for(int j=y;j>0;j-=lowbit(j))
            ans+=c[i][j];
    return ans;
}
int main()
{
//    freopen("in.txt","r",stdin);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        memset(c,0,sizeof(c));
        char op[5];
        scanf("%d%d",&n,&m);
        for(int i=0;i<m;i++)
        {
            scanf("%s",op);
            if(op[0]=='C')
            {
                scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
                range_add(x1,x2,y1,y2,1);
            }
            else
            {

                scanf("%d%d",&x,&y);
                int ans=sum(x,y);
                if(ans%2==1)
                    printf("1\n");
                else
                    printf("0\n");
            }
        }
        if(T) printf("\n");//注意输出格式
    }
    return 0;
}