WC 2005 dface 双面棋盘

来源:互联网 发布:怎么判断usb2.0端口 编辑:程序博客网 时间:2024/04/26 06:34

其实就是个线段树+并查集维护,原来的每行看做线段树的一个底层节点,线段树每个节点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;}


原创粉丝点击