熄灯问题（枚举）

描述

有一个由按钮组成的矩阵，其中每行有6个按钮，共5行。每个按钮的位置上有一盏灯。当按下一个按钮后，该按钮以及周围位置(上边、下边、左边、右边)的灯都会改变一次。即，如果灯原来是点亮的，就会被熄灭；如果灯原来是熄灭的，则会被点亮。在矩阵角上的按钮改变3盏灯的状态；在矩阵边上的按钮改变4盏灯的状态；其他的按钮改变5盏灯的状态。

在上图中，左边矩阵中用X标记的按钮表示被按下，右边的矩阵表示灯状态的改变。对矩阵中的每盏灯设置一个初始状态。请你按按钮，直至每一盏等都熄灭。与一盏灯毗邻的多个按钮被按下时，一个操作会抵消另一次操作的结果。在下图中，第2行第3、5列的按钮都被按下，因此第2行、第4列的灯的状态就不改变。

请你写一个程序，确定需要按下哪些按钮，恰好使得所有的灯都熄灭。根据上面的规则，我们知道1）第2次按下同一个按钮时，将抵消第1次按下时所产生的结果。因此，每个按钮最多只需要按下一次；2）各个按钮被按下的顺序对最终的结果没有影响；3）对第1行中每盏点亮的灯，按下第2行对应的按钮，就可以熄灭第1行的全部灯。如此重复下去，可以熄灭第1、2、3、4行的全部灯。同样，按下第1、2、3、4、5列的按钮，可以熄灭前5列的灯。

输入

5行组成，每一行包括6个数字（0或1）。

相邻两个数字之间用单个空格隔开。

0表示灯的初始状态是熄灭的，1表示灯的初始状态是点亮的。

输出

5行组成，每一行包括6个数字（0或1）。

相邻两个数字之间用单个空格隔开。

其中的1表示需要把对应的按钮按下，0则表示不需要按对应的按钮。

输入样例 1 

2
0 1 1 0 1 0
1 0 0 1 1 1
0 0 1 0 0 1
1 0 0 1 0 1
0 1 1 1 0 0
0 0 1 0 1 0
1 0 1 0 1 1
0 0 1 0 1 1
1 0 1 1 0 0
0 1 0 1 0 0

输出样例：

PUZZLE #1
1 0 1 0 0 1 
1 1 0 1 0 1 
0 0 1 0 1 1 
1 0 0 1 0 0 
0 1 0 0 0 0 
PUZZLE #2
1 0 0 1 1 1 
1 1 0 0 0 0 
0 0 0 1 0 0 
1 1 0 1 0 1 
1 0 1 1 0 1 

注意：PUZZLE行结尾没有空格，数字行最后有一个空格。

思路：

当第一排的灯的状态确定后，关灯的方式是唯一的，所以先枚举出第一排灯的所有状态可能，在进行验证。

代码实现：

#include<iostream>
#include<string.h>
using namespace std;
bool lamp[5][6]={0};
bool res[5][6]={0};
int dx[4]={-1,0,1,0};
int dy[4]={0,-1,0,1};
bool judge(bool temp[][6])
{
    for(int i=0;i<5;i++)
        for(int j=0;j<6;j++)if(temp[i][j])return false;
    return true;
}
void print_res()
{
    for(int i=0;i<5;i++)
    {
        for(int j=0;j<6;j++)
        {
            cout<<res[i][j]<<" ";
        }
        cout<<endl;
    }
}
void button(int x,int y,bool temp[][6])
{
    if(temp[x][y])temp[x][y]=0;
    else temp[x][y]=1;
}
void operate(int x,int y,bool temp[][6])
{
    button(x,y,temp);
    for(int i=0;i<4;i++)
    {
        int tx=x+dx[i],ty=y+dy[i];
        if(tx>=0&&ty>=0&&tx<5&&ty<6)button(tx,ty,temp);
    }
}
bool xideng(bool temp[][6]);
bool enumerate(int);
int main()
{
    int n;
    cin>>n;
    for(int k=1;k<=n;k++)
    {
        for(int i=0;i<5;i++)
            for(int j=0;j<6;j++)
        {
            cin>>lamp[i][j];
        }
        memset(res,0,sizeof(res));
        cout<<"PUZZLE #"<<k<<endl;
        enumerate(0);
    }
    return 0;
}
bool xideng(bool temp[][6])
{
    for(int i=1;i<5;i++)
        for(int j=0;j<6;j++)res[i][j]=0;
    for(int i=0;i<4;i++)
    {
        for(int j=0;j<6;j++)
        {
            if(temp[i][j])
            {
                operate(i+1,j,temp);
                res[i+1][j]=1;
            }
        }
    }
    if(judge(temp))
    {
        print_res();
        return true;
    }
    return false;
}
bool enumerate(int u)
{
    if(u==6)
    {
        bool temp[5][6]={0};
        for(int i=0;i<5;i++)
            for(int j=0;j<6;j++)temp[i][j]=lamp[i][j];
        if(xideng(temp))return true;
        else return false;
    }
    if(enumerate(u+1))return true;
    operate(0,u,lamp);
    res[0][u]=1;
    if(enumerate(u+1))return true;
    operate(0,u,lamp);
    res[0][u]=0;
    return false;
}