OpenJudge:熄灯问题

题目链接
大意就是，摁一个开关，它的前后左右以及他自己的状态都会改变。
原本是开的变成关的，原本是关的变成开的。
我们的任务就是把所有的灯都变成关闭状态。
我们可以这样想象，第一排的灯已近摁完了，确实是正确答案，第一行就不再摁了，于是，要让第一排的灯全部熄灭，就需要通过摁第二行的灯，以此下去后面的灯的状态也就确定了。

#include<iostream>
#include<cstring>
using namespace std;
const int ax[5] = {1, 0, 0, -1, 0};//摁灯的五个点。
const int ay[5] = {0, 1, -1, 0, 0};
int maze[10][10], backup[10][10], ans[10][10];
void turn(int x, int y) {//反转灯的状态。
	for(int i = 0; i < 5; i++) {
		int tempx = x + ax[i];
		int tempy = y + ay[i];
		if(tempx >= 0 && tempx < 5 && tempy >= 0 && tempy < 6)//是否合理，防止发生数组越界。
			maze[tempx][tempy] = !maze[tempx][tempy];//取反。
	}
}
void solve() {
	memcpy(backup, maze, sizeof maze);//先备份原来一开始的状态，方便后面枚举。
	for(int i = 0; i < 1 << 6; i++) {//总共有0 ~ (1^6 - 1)种状态。
		memset(ans, 0, sizeof ans);//清空错误的ans数组，
		for(int j = 0; j < 6; j++)//如果第j位是1 代表摁过，需要反转与其相关的灯。
			if(i >> j & 1) {
				ans[0][j] = 1;
				turn(0, j);
			}
		for(int j = 0; j < 4; j ++)//通过摁后一排的灯来保证前面的灯是合理的状态。
			for(int k = 0; k < 6; k++)
				if(maze[j][k] == 1) {
					ans[j + 1][k] = 1;
					turn(j + 1, k);
				}
		int flag = 1;
		for(int j = 0; j < 6; j++)//前面的灯已近保证是合理的状态了，
			if(maze[4][j] == 1) {//检查最后一排的灯是否合理
				flag = 0;
				break;
			}
		if(flag )	break;//合理直接输出答案。
		memcpy(maze, backup, sizeof backup);//还原原来的状态，再继续枚举找答案。
	}
	for(int i = 0; i < 5; i++) {//得到答案输出。
		for(int j = 0; j < 6; j++)
			cout << ans[i][j] << " ";
		cout << endl;
	}
}
int main() {
//	freopen("D:\MY\answer.txt","r",stdin);
	for(int i = 0; i < 5; i++)
		for(int j = 0; j < 6; j++)
			cin >> maze[i][j];
	solve();
	return 0;
}