WC 2005 dface 双面棋盘

其实就是个线段树+并查集维护，原来的每行看做线段树的一个底层节点，线段树每个节点2*n的空间建立一个并查集，两个节点合并时维护块的个数信息，然后舍弃中间部分保留外围部分重建并查集就可以了。

我写的是zkw线段树，用这方法就算不是黑白双色也可以的吧。

本人弱菜。

#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
int sum[517][5],a[517][517],ll[517],rr[517],f[517][805];
int n=0,mm=0,tot=0;
int find(int s,int x)//在s号并查集中find和路压
{
  if (f[s][x]==x) return x;
  f[s][x]=find(s,f[s][x]);
  return f[s][x];
}
void search(int p,int s)//底层节点特别维护
{
  int i=0,recent=0;
  ll[s]=rr[s]=p;
  sum[s][0]=0;
  sum[s][1]=0;
  for (i=1;i<=n;i++)
    {
      if (a[p][i]==a[p][i-1])
        f[s][i]=recent;
      else
        {
          sum[s][a[p][i]]++;
          recent=i;
          f[s][i]=i;
        }
    }
}
void merge(int s,int l,int r)//合并两个节点l,r。信息传给s
{
  int i=0,fx=0,fy=0;
  memset(f[s],0,sizeof(f[s]));
  ll[s]=ll[l];
  rr[s]=rr[r];
  sum[s][0]=sum[l][0]+sum[r][0];
  sum[s][1]=sum[l][1]+sum[r][1];
  sum[s][2]=0;
  if (ll[l]==rr[l])//l,r为底层节点
    {
      for (i=1;i<=n;i++)
        {
          f[s][i]=f[l][i];
          f[s][i+n]=f[r][i]+n;
        }
      for (i=1;i<=n;i++)
        if (a[rr[l]][i]==a[ll[r]][i])
          {
            fx=find(s,i);
            fy=find(s,i+n);
            if (fx!=fy)
              {
                f[s][fx]=fy;
                sum[s][a[rr[l]][i]]--;
              }
          }
      for (i=1;i<=n;i++)
        f[s][i]=find(s,i);
    }
  else
    {
      for (i=1;i<=n;i++)
        {
          f[s][i]=f[l][i];
          f[s][i+n]=f[l][i+n];
          f[s][i+2*n]=f[r][i]+2*n;
          f[s][i+3*n]=f[r][i+n]+2*n;
        }
      for (i=1;i<=n;i++)
        if (a[rr[l]][i]==a[ll[r]][i])
          {
            fx=find(s,i+n);
            fy=find(s,i+2*n);
            if (fx!=fy)
              {
                f[s][fx]=fy;
                sum[s][a[rr[l]][i]]--;
              }
          }//以下为重建并查集
      for (i=1;i<=n;i++)
        {
          fx=find(s,i);
          if (n<fx && fx<=3*n)
            {
              f[s][fx]=i;
              f[s][i]=i;
            }
        }
      for (i=3*n+1;i<=4*n;i++)
        {
          fx=find(s,i);
          if (n<fx && fx<=3*n)
            {
              f[s][fx]=i;
              f[s][i]=i;
            }
        }
      for (i=1;i<=n;i++)
        {
          if (3*n<f[s][i] && f[s][i]<=4*n)
            f[s][i]-=2*n;
          f[s][i+n]=f[s][i+3*n];
          if (3*n<f[s][i+n] && f[s][i+n]<=4*n)
            f[s][i+n]-=2*n;
        }
    }
  sum[s][2]=0;
  sum[s][3]=0;//为了便于维护,在边界上填充2,3的颜色,与0,1区别
}
void modify(int now)
{
  search(now,now+mm);
  now+=mm;
  now>>=1;
  while (now)
    {
      merge(now,now<<1,(now<<1)+1);
      now>>=1;
    }
}
int main()
{
  freopen("dface.in","r",stdin);
  freopen("dface.out","w",stdout);
  scanf("%d",&n);
  int i=0,j=0,x=0,y=0;
  for (i=1;i<=n;i++)
    {
      for (j=1;j<=n;j++)
        scanf("%d",&a[i][j]);
    }
  mm=1<<(int)(log2(n)+1);
  if (mm<=n) mm<<=1;
  for (i=0;i<mm;i++)
    a[i][0]=3;
  for (i=1;i<=n;i++)
    a[0][i]=2;
  for (i=n+1;i<mm;i++)
    for (j=1;j<=n;j++)
      a[i][j]=2;
  for (i=0;i<mm;i++)
    search(i,i+mm);
  for (i=mm-1;i>=1;i--)
    merge(i,i<<1,(i<<1)+1);
  int m=0;
  scanf("%d\n",&m);
  for (;m>0;m--)
    {
      scanf("%d%d",&x,&y);
      a[x][y]=a[x][y]^1;
      modify(x);
      printf("%d %d\n",sum[1][1],sum[1][0]);
    }
  return 0;
}