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;}