题意:一个5*6的灯方阵,0表示灯灭,1表示灯亮。按一盏灯其四邻域及其自身的颜色会发生改变,求出一种方法使得所有灯全灭。
思路:枚举第一行的6个灯,共64种情况。第一行确定后以后每行的关灯情况就确定了,最后判断最后一行的灯光是否全灭即可。
#include <stdio.h>
#include <string.h>
#include <math.h>
#define N 8
int S,T;
int s[N][N],t[N][N],res[N][N];//t是每个位置灯亮灭的变换次数
void myfill(int x){
int i;
for(i = 1;i<=6;i++){
res[1][7-i] = x&1;
x>>=1;
}
}
void update(int d){
int i;
for(i = 1;i<=6;i++)
if(res[d][i]){
t[d][i]++;
t[d-1][i]++;
t[d+1][i]++;
t[d][i-1]++;
t[d][i+1]++;
}
}
int finish(){
int i;
for(i = 1;i<=6;i++)
if((s[5][i]+t[5][i])&1)
return 0;
return 1;
}
int main(){
freopen("a.txt","r",stdin);
scanf("%d",&T);
for(S = 1;S<=T;S++){
int i,j,k;
for(i = 1;i<=5;i++)
for(j = 1;j<=6;j++)
scanf("%d",&s[i][j]);
for(k = 0;k<(1<<6);k++){//枚举第一行
memset(t,0,sizeof(t));
memset(res,0,sizeof(res));
myfill(k);//填充第一行
update(1);
for(i = 2;i<=5;i++){
for(j = 1;j<=6;j++)
if((s[i-1][j]+t[i-1][j])&1)
res[i][j] = 1;
update(i);//将按下灯的四邻域位置亮灭次数更新
}
if(finish())
break;
}
printf("PUZZLE #%d\n",S);
for(i = 1;i<=5;i++){
for(j = 1;j<=6;j++)
printf("%d ",res[i][j]);
putchar('\n');
}
}
return 0;
}