校赛预选卡了一道高斯消元的题,找到资料学习了下,a个模板题熟练熟练。写错一个数字坑了好久,悲催。
题意:给定5*6的灯阵,0为关1为开,每次开关(x,y)会同时开关(x-1,y)(x+1,y)(x,y-1)(x,y+1)四灯(出界不予讨论即可),给定初始灯阵,求使所有灯都关闭的开灯方法。
分析:从上往下从左往右计数,第i个灯的开关情况记为ai,ai=0不动,ai=1开一次。每个灯的最终状态由它和它周围的灯开关状态确定。
对每个灯建立30元一次异或方程;
例如:
0 1 1 0 1 0
1 0 0 1 1 1
0 0 1 0 0 1
1 0 0 1 0 1
0 1 1 1 0 0
第1个方程为a1^a2^a7=0
第2个方程为a1^a2^a3^a8=1
以此类推建立30元1次异或方程组,解之即可。
解方程用高斯消元,不多说上代码。
#include <cstdio>
#include <cstring>
using namespace std;
int jz[31][32];
void swap(int a,int b)
{
for(int i=1;i<=31;i++)
{
int t=jz[a][i];
jz[a][i]=jz[b][i];
jz[b][i]=t;
}
return;
}
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
int tmp;
memset(jz,0,sizeof(jz));
for(int x=1;x<=5;x++)
for(int y=1;y<=6;y++){
scanf("%d",&tmp);
int po=(x-1)*6+y;
jz[po][31]=tmp;
jz[po][po]=1;
if(x>1)jz[po][po-6]=1;
if(x<5)jz[po][po+6]=1;
if(y>1)jz[po][po-1]=1;
if(y<6)jz[po][po+1]=1;
}
/*for(int x=1;x<=30;x++)
{
for(int y=1;y<=31;y++)printf("%d ",jz[x][y]);
printf("\n");
}
printf("\n");*/
for(int x=1;x<=30;x++)
{
int po=x,ok=0;
for(int x1=x;x1<=30;x1++)
{
if(jz[x1][x]){po=x1;ok=1;break;
}
}
if(!ok)continue;
swap(x,po);
for(int x1=1;x1<=30;x1++)
{
if(jz[x1][x]&&x1!=x)
{
for(int y=1;y<=31;y++)
jz[x1][y]^=jz[x][y];
}
}
}
/*for(int x=1;x<=30;x++)
{
for(int y=1;y<=31;y++)printf("%d ",jz[x][y]);
printf("\n");
}*/
printf("PUZZLE #%d\n",i);
for(int x=1;x<=5;x++)
{
for(int y=1;y<=5;y++)
{
printf("%d ",jz[(x-1)*6+y][31]);
}
printf("%d\n",jz[6*x][31]);
}
}
return 0;
}