POJ-1222 EXTENDED LIGHTS OUT【暴力枚举】

POJ-1222 EXTENDED LIGHTS OUT【暴力枚举】

题目链接：http://poj.org/problem?id=1222

题意：（开关问题）5*6的矩阵，按下一个灯其上下左右的灯随之变化，求如何操作使得灯都熄灭

**题解：**5*6的数据很小，可以暴力枚举情况。首先枚举第一行的所有情况2^6 = 64 种情况（000000、100000、010000、110000……）再根据上一个行的状态处理，最后根据最后一行的状态判断

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <map>
#include <cmath>
using namespace std;
#define eps 1e-6
#define pi 3.14159265359
typedef long long LL;
typedef long double LD;
const int maxn = 10;
int n,m,t;
int a[maxn][maxn],b[maxn][maxn];
int flag[maxn][maxn];
int dir[4][2]={0,1,0,-1, 1,0,-1,0}; // 方向向量，(x,y)周围的四个方向
int Check(int x,int y){//判断条件
if( x>=1 && x<=n && y>=1 && y<=m ) return 1;
else return 0;
}

void pre(int x,int y){//按键操作
a[x][y] = !a[x][y]; //逻辑非
for(int i=0;i<4;i++){
int xx = x+dir[i][0],yy = y+dir[i][1];
if(Check(xx,yy) ){
a[xx][yy] = !a[xx][yy];
}
}
}
void fun(){
int _flag = 0;
int k = 1;
while(k<=64){//第一行开关情况：2^6种
memcpy(a,b,sizeof(b));
memset(flag,0,sizeof(flag));
for(int i=0;i<=5;i++){//列举第一行所有开关情况
if(k&(1<<i) ){
pre(1,1+i);
flag[1][1+i] = 1;
}
}
for(int i=2;i<=n;i++){
for(int j=1;j<=m;j++){
if(a[i-1][j]){
pre(i,j);
flag[i][j] = 1;
}
}
}
int flag_ = 1;
for(int i=1;i<=m;i++){
if(a[5][i]){//不为0不满足
flag_ = 0;
break;
}
}
if(flag_){
for(int i=1; i<=n; i++){
for(int j=1; j<m; j++)
printf("%d ",flag[i][j]);
printf("%d\n",flag[i][6]);
}
break;
}
k++;
}
}

int main(){
scanf("%d",&t);
n=5,m=6;
for(int k=1;k<=t;k++){
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
scanf("%d",&b[i][j]);
}
}
printf("PUZZLE #%d\n",k);
fun();
}
return  0;
}