POJ1221熄灯问题

https://vjudge.net/problem/POJ-1222
思路：枚举第一行。从第二行开始，如果上边一行的等还是开的，就按这个开关，直到最后一行。检查等是否全灭掉了，若是就输出按开关的方案。
（位运算用的不习惯，就直接用数组了）

#include <iostream>
#include <cstring>
using namespace std;
const int N = 10;
int t, light[N][N], cnt;
int a[N][N], state = (1 << 6) - 1, op[N][N];
//这个灯取反 
void Flip(int i, int j){
	if(a[i][j]) a[i][j] = 0;
	else a[i][j] = 1;
}
//改变周围灯的状态 
void change(int i, int j){
	Flip(i, j);
	Flip(i + 1,j);
	Flip(i - 1, j);
	Flip(i, j + 1);
	Flip(i, j - 1);
}
//检查是否全灭掉了 
bool check(){
	for(int i = 1;i <= 5; ++ i)
		for(int j = 1;j <= 6; ++ j)
			if(a[i][j]) return false;
	return true;
}
void Print(){
	for(int i = 1;i <= 5; ++ i)
	{
		for(int j = 1;j <= 6; ++ j)
		printf("%d ",op[i][j]);
		puts("");
	}
}
int main()
{
	cin >> t;
	while(t--){
		
		for(int i = 1;i <= 5; ++ i)
			for(int j = 1;j <= 6; ++ j)
				cin >> light[i][j];
		
		for(int k = 0;k <= state; ++ k){
			int l = k;
			memset(op, 0, sizeof(op));
			for(int i = 6;i >= 1; -- i)
				op[1][i] = (l & 1), l >>= 1;
				
			for(int i = 1;i <= 5; ++ i)
				for(int j = 1;j <= 6; ++ j)
					a[i][j] = light[i][j];
				
			for(int j = 1;j <= 6; ++ j)
				if(op[1][j]){
					change(1, j);
				}
			//检查上一行的灯是否灭掉 
			for(int i = 2;i <= 5; ++ i){
				for(int j = 1;j <= 6; ++ j){
					if(a[i-1][j])
					   change(i, j), op[i][j] = 1;
				}
			}
			if(check()){
				printf("PUZZLE #%d\n", ++ cnt);
				Print();
				break;
			}
		}	
	}
}