2811 熄灯问题

熄灯问题

熄灯问题
观察题目，固定矩阵，且规模为5*6，很小，可以使用枚举
有题意得：对第1行中每盏点亮的灯，按下第2行对应的按钮，就可以熄灭第1行的全部灯
所以第一行的等亮状态就决定了后面四行的按钮按压情况，所以枚举第一行按压状态，使第五行剩余灯全灭时，就输出答案
code

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<vector>
#include<map>
#include<queue>
#include<set>
typedef long long LL;
using namespace std;
int a[6][7],book[6][7],tmp[6][7];
void reverse(int x,int y){
	book[x][y]=1;
	tmp[x][y]=1-tmp[x][y];
	tmp[x-1][y]=1-tmp[x-1][y];
	tmp[x+1][y]=1-tmp[x+1][y];
	tmp[x][y-1]=1-tmp[x][y-1];
	tmp[x][y+1]=1-tmp[x][y+1];
}
bool check(){
	for(int i=1;i<=6;i++)
		if(tmp[5][i])	return false;
	return true;
}
int main(){
	for(int i=1;i<=5;i++)
		for(int j=1;j<=6;j++)
			scanf("%d",&a[i][j]);
	for(int k=0;k<=63;k++){
		for(int i=1;i<=5;i++)
			for(int j=1;j<=6;j++)
				book[i][j]=0;
		//复制 
		for(int i=1;i<=5;i++)
			for(int j=1;j<=6;j++)
				tmp[i][j]=a[i][j];
		for(int i=0;i<=5;i++){
			if(k&(1<<i)){
				reverse(1,i+1);
			}
		}
		//开始测试 
		for(int i=1;i<=4;i++){
			for(int j=1;j<=6;j++){
				if(tmp[i][j])	reverse(i+1,j);
			}
		}
		//检查 
		if(check()){
			for(int i=1;i<=5;i++){
				for(int j=1;j<=6;j++)
					printf("%d ",book[i][j]);
				printf("\n");
			}
//			for(int i=1;i<=5;i++){
//				for(int j=1;j<=6;j++){
//					printf("%d ",tmp[i][j]); 
//				}
//				printf("\n");
//			} 
			return 0;
		}
	}	
}