BZOJ1453: [Wc]Dface双面棋盘

题目

http://www.lydsy.com/JudgeOnline/problem.php?id=1453

题解

线段树上维护奇怪的信息第二题，这道题写的很快，1h，但是因为合并的时候把并查集合并写错了qwqqqq卧槽啊这个时候了我居然还在写错这些、记住是修改根的父亲为根啊（倒）。

思路还是挺简单啦，每个节点开两个200的数组记录两边的连通信息，在一个连通块的标号一样，然后合并的时候，交界处如果颜色相同就认为有一条边然后做并查集合并。并查集也是真的厉害辣，代码很简单啦（可能是被上一道题虐傻了）。

代码

//QWsin
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=200+10;

int col[maxn][maxn];

int p[maxn*4],id[maxn*4];
int findset(int x){return p[x]==x?x:p[x]=findset(p[x]);}

int n;
struct G{//存放连通性 
    int l,r;
    int cnt1,cnt2,T,pl[maxn],pr[maxn];
    //维护连通块个数，pl左pr右并查集,T记录边界上并查集编号最大值（方便合并） 
    //cnt1 black cnt2 white
    G(int l=0,int r=0):l(l),r(r){
        cnt1=cnt2=0;memset(pl,0,sizeof pl);
        memset(pr,0,sizeof pr);
    }

    G operator + (const G &rhs)const{
        G ret(l,rhs.r);

        int C1=cnt1+rhs.cnt1;
        int C2=cnt2+rhs.cnt2;
        for(int i=1;i<=T+rhs.T;++i) p[i]=i,id[i]=0;

        //右边的连通块编号统一加上左边的T值，合并为一个图再连边
        for(int i=1;i<=n;++i) if(col[i][r]==col[i][r+1])
        {
            int p1=findset(pr[i]);
            int p2=findset(rhs.pl[i]+T);
            if(p1!=p2){
                p[p2]=p1;//原来把并查集这里合并写错了(捂脸)
                if(col[i][r]) --C1;
                else --C2;
            }
        }
        //离散一下编号，因为都不大所以可以直接用数组记离散后对应的编号
        int rank=0;ret.cnt1=C1;ret.cnt2=C2;
        for(int i=1;i<=n;++i){
            int p1=findset(pl[i]);
            int p2=findset(rhs.pr[i]+T);
            if(!id[p1]){id[p1]=++rank;}
            if(!id[p2]){id[p2]=++rank;}
            ret.pl[i]=id[p1];
            ret.pr[i]=id[p2];
        }
        ret.T=rank;

        return ret;
    }

    //单独更新pos列的信息 
    inline void work(int pos)
    {
        cnt1=cnt2=0;
        for(int i=1;i<=n;++i)
        {
            if(i==1||col[i][pos]!=col[i-1][pos])
            {
                if(col[i][pos]) ++cnt1;
                else ++cnt2;
            }
            pl[i]=pr[i]=cnt1+cnt2;
        }
        T=cnt1+cnt2;
    }
};

struct Node{
    G g;
    Node *lc,*rc;   
    Node(int l,int r){lc=rc=NULL;g=G(l,r);}
    inline void up(){g=lc->g+rc->g;}
}*root;

#define mid ((l+r)>>1)
void build(Node* &p,int l,int r)
{
    p=new Node(l,r);
    if(l==r) {p->g.work(l);return ;}
    build(p->lc,l,mid);
    build(p->rc,mid+1,r);
    p->up();
}

void updata(Node* p,int l,int r,int pos)
{
    if(l==r){p->g.work(l);return ;}
    if(pos<=mid) updata(p->lc,l,mid,pos);
    else updata(p->rc,mid+1,r,pos); 
    p->up();
}

int main()
{
    cin>>n;
    for(int i=1;i<=n;++i)
        for(int j=1;j<=n;++j)
            scanf("%d",col[i]+j);

    build(root,1,n);
    int m,a,b;cin>>m;   
    while(m--)
    {
        scanf("%d%d",&a,&b);
        col[a][b]^=1;
        updata(root,1,n,b);
        printf("%d %d\n",root->g.cnt1,root->g.cnt2);//很良心的一点就是不用写query了
    }
    return 0;
}