1222:EXTENDED LIGHTS OUT

枚举

解题思路：

1. 枚举第一行的翻转策略。
2. 然后从第二行开始分析，如果( i − 1 , j ) 位置是亮的，那么( i , j )位置就必须翻转，只有他能影响到( i − 1 , j )。
3. 反复判断直到最后一行，如果最后一行能全关，就是答案。

参考：
https://blog.csdn.net/csyifanZhang/article/details/104560695

//枚举，找可行解
#include <iostream>
#include <cstring>
using namespace std;

const int N = 10;

int arr[N][N];//初始状态
int ans[N][N];//结果数组
int flip[N][N];//中间状态
int dir[5][2] = { {1,0},{-1,0},{0,0},{0,1},{0,-1} };

int judge(int x, int y) {//判断该位置灯的状态
	int i, c = arr[x][y],xx,yy;
	for (i = 0; i < 5; i++) {//遍历周围的灯
		xx = x + dir[i][0];
		yy = y + dir[i][1];
		if (xx >= 0 && xx < 5 && y >= 0 && y < 6)
			c += flip[xx][yy];
	}
	return c % 2;//偶数，灭；奇数：亮
}

int calc() {
	int i, j;
	for (i = 1; i < 5; i++) {
		for (j = 0; j < 6; j++) {
			if (judge(i - 1, j) != 0)//上方格子是1（亮着），则必须翻转(i,j)号格子
				flip[i][j] = 1;
		}
	}
	for (i = 0; i < 6; i++) {//判断最后一行是否全0
		if (judge(4, i) != 0)//当前方案不可行
			return -1;
	}
	return 1;
}

int main() {
	int t, i, j,num,k=1;
	cin >> t;
	while(t--) {
		num = -1;
		for (i = 0; i < 5; i++) {
			for (j = 0; j < 6; j++)
				cin >> arr[i][j];
		}
		//按照字典序枚举第一行所有的可能，方案数为2^6 
		for (i = 0; i < 1 << 6; i++) {//i化为二进制就是本次第一行的翻转方式
			memset(flip, 0, sizeof(flip));
			for (j = 0; j < 6; j++)
				//第一行初始化（从右到左），移位后和1与运算，表示[0][5-j]位置的灯是否翻转
				flip[0][5 - j] = i >> j & 1;
			num = calc();
			if (num > 0)
				memcpy(ans, flip, sizeof(flip));//复制
		}
		printf("PUZZLE #%d\n", k++);
		for (i = 0; i < 5; i++) {
			for (j = 0; j < 5; j++)
				cout << ans[i][j] << " ";
			cout << ans[i][j] << endl;
		}
	}
	return 0;
}