bzoj 1453 双面棋盘 LCT 并查集

把每个格子看成点，同色格子之间连边。
LCT维护关于删除时间的最大生成树。
设有x1个原来与当前格子同色的连通块与当前格子断开，x2个原来与当前格子不同色的连通块与当前格子连通，那么会增加x1-1个当前格子原来颜色的连通块，减少x2-1个当前格子现在颜色的连通块。
写得比较挫，初始状态用的并查集。。。

#include <bits/stdc++.h>
using namespace std;
#define N 210
#define M 310000
#define which(x) (ch[fa[x]][1]==x)
int n,m,cnt,tot,ans[2];
int a[N][N],idn[N][N],ide[N][N][2],idp[M][2];
int tim[M],pos[M],del[M],tmp[M],tp[M],val[M],fat[M];
int ch[M][2],fa[M],rev[M],mn[M],pos1[M],use[M],opt[M],opp[M];
void ins(int x,int y,int type,int op,int x1,int y1)
{tim[++tot]=x;pos[tot]=y;tp[tot]=type;opt[tot]=op;opp[tot]=idn[x1][y1];}
int find1(int x){return fat[x]==x ? x:fat[x]=find1(fat[x]);}
void add(int x1,int y1,int x2,int y2,int tp)
{
    ide[x1][y1][tp]=++cnt;
    idp[cnt][0]=idn[x1][y1];
    idp[cnt][1]=idn[x2][y2];
    if(a[x1][y1]==a[x2][y2])
    {
        ins(0,ide[x1][y1][tp],1,1,x1,y1);
        fat[find1(idn[x1][y1])]=find1(idn[x2][y2]);
    }
}
void proc(int x)
    {swap(ch[x][0],ch[x][1]);rev[x]^=1;}
void pushdown(int x)
{
    if(rev[x])
    {
        proc(ch[x][0]);proc(ch[x][1]);
        rev[x]=0;
    }
}
int isroot(int x)
    {return !fa[x]||ch[fa[x]][which(x)]!=x;}
void down(int x)
{
    if(!isroot(x))down(fa[x]);
    pushdown(x);
}
void pushup(int x)
{
    pos1[x]=mn[ch[x][0]]<mn[ch[x][1]] ? 
    (mn[ch[x][0]]<val[x] ? pos1[ch[x][0]]:x):
    (mn[ch[x][1]]<val[x] ? pos1[ch[x][1]]:x); 
    mn[x]=min(val[x],min(mn[ch[x][0]],mn[ch[x][1]]));
}
void rotate(int x)
{
    int y=fa[x],k=which(x);
    ch[y][k]=ch[x][k^1];
    ch[x][k^1]=y;
    if(!isroot(y))ch[fa[y]][which(y)]=x;
    fa[x]=fa[y];fa[y]=x;
    fa[ch[y][k]]=y;
    pushup(y);pushup(x);
}
void splay(int x)
{
    down(x);
    while(!isroot(x))
    {
        int y=fa[x];
        if(isroot(y))rotate(x);
        else
        {
            if(which(x)^which(y))rotate(x);
            else rotate(y);
            rotate(x);
        }
    }
}
void access(int x)
{
    int t=0;
    while(x)
    {
        splay(x);
        ch[x][1]=t;fa[t]=x;
        pushup(x);
        t=x;x=fa[x];
    }
}
void rever(int x)
{
    access(x);splay(x);
    proc(x);
}
void link(int x,int y)
{
    rever(x);fa[x]=y;
}
void cut(int x,int y)
{
    rever(x);
    access(y);splay(y);
    ch[y][0]=fa[x]=0;
    pushup(x);
}
int find(int x)
{
    while(fa[x])x=fa[x];
    return x;
}
int main()
{
    //freopen("tt.in","r",stdin);
    scanf("%d",&n);
    if(n==1)
    {
        int x;scanf("%d%d",&x,&m);
        for(int i=1;i<=m;i++)
            printf("%d %d\n",x^1,x),x^=1;
        return 0;
    }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        {
            scanf("%d",&a[i][j]);
            idn[i][j]=++cnt;fat[cnt]=cnt;
            if(i!=1)add(i,j,i-1,j,0);
            if(j!=1)add(i,j,i,j-1,1);
        }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            if(find1(idn[i][j])==idn[i][j])
                ans[a[i][j]]++;
    scanf("%d",&m);
    for(int i=1,x,y,t;i<=m;i++)
    {
        scanf("%d%d",&x,&y);a[x][y]^=1;t=a[x][y];
        if(x!=1)ins(i,ide[x][y][0],a[x][y]==a[x-1][y],t,x,y);
        if(y!=1)ins(i,ide[x][y][1],a[x][y]==a[x][y-1],t,x,y);
        if(x!=n)ins(i,ide[x+1][y][0],a[x][y]==a[x+1][y],t,x,y);
        if(y!=n)ins(i,ide[x][y+1][1],a[x][y]==a[x][y+1],t,x,y);
    }
    memset(tmp,0x3f,sizeof(tmp));
    for(int i=tot;i>=1;i--)
    {
        if(!tp[i])tmp[pos[i]]=tim[i];
        else del[i]=tmp[pos[i]];
    }
    memset(val,0x3f,sizeof(val));
    memset(mn,0x3f,sizeof(mn));
    for(int i=1,now=1;i<=m;i++)
    {
        int n1=0,n2=0,vt;
        while(now<=tot&&tim[now]<=i)
        {
            int t=pos[now];
            rever(opp[now]);
            if(tp[now])
            {
                val[t]=del[now];mn[t]=del[now];
                if(find(idp[t][0])==find(idp[t][1]))
                {
                    rever(idp[t][0]);
                    access(idp[t][1]);splay(idp[t][1]);
                    if(mn[idp[t][1]]<val[t])
                    {
                        int t1=pos1[idp[t][1]];
                        use[t1]=0;use[t]=1;
                        cut(t1,idp[t1][0]);cut(t1,idp[t1][1]);
                        link(t,idp[t][0]);link(t,idp[t][1]);
                    }
                }
                else
                {
                    link(t,idp[t][0]);link(t,idp[t][1]);
                    use[t]=1;if(tim[now])n1++;
                }
            }
            else if(use[t])
            {
                use[t]=0;
                cut(t,idp[t][0]);cut(t,idp[t][1]);
                if(tim[now])n2++;
            }
            vt=opt[now];now++;
        }
        ans[vt]-=n1-1,ans[vt^1]+=n2-1;
        printf("%d %d\n",ans[1],ans[0]);
    }
    return 0;
}