poj 1222 枚举(关灯游戏)

题意：一个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;
}