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