POJ熄灯问题（枚举法）

枚举法练习：

题目：

有一个由按钮组成的矩阵，其中每行有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列的灯。

输入

第一行是一个正整数N，表示需要解决的案例数。每个案例由5行组成，每一行包括6个数字。这些数字以空格隔开，可以是0或1。0表示灯的初始状态是熄灭的，1表示灯的初始状态是点亮的。

输出

对每个案例，首先输出一行，输出字符串“PUZZLE #m”，其中m是该案例的序号。接着按照该案例的输入格式输出5行，其中的1表示需要把对应的按钮按下，0则表示不需要按对应的按钮。每个数字以一个空格隔开。

样例输入

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 

代码：

#include <iostream>
#include <cstring>
using namespace std;


bool puzzle[ 6 ][ 7 ], press[ 6 ][ 7 ], res[ 6 ][ 7 ];
void Change( int i,int j )
{
    press[ i ][ j ] = !press[ i ][ j ];
    press[ i ][ j - 1 ] = !press[ i ][ j - 1 ];
    press[ i ][ j + 1 ] = !press[ i ][ j + 1 ];
    press[ i - 1 ][ j ] = !press[ i - 1 ][ j ];
    press[ i + 1 ][ j ] = !press[ i + 1 ][ j ];
}

bool PressSuccessfully()
{
    for( int i = 1; i <= 6; i++ )
    {
        if( res[ 1 ][ i ] == 1)
            Change( 1, i );
    }
    for( int i = 2; i <=5; i++ )
    {
        for( int j = 1; j <= 6; j++ )
            if( press[ i - 1 ][ j ] == 1 )
            {
                res[ i  ][ j ] = 1;
                Change( i  , j );
            }

    }
    for( int i = 1; i <= 6; i++ )
        if( press[ 5 ][ i ] == 1 )
        {
            return false;
        }
    return true;
}

int SelectPress( int num )
{
    for( int a = 0; a <= 1; a++ )
        for( int b = 0; b <= 1; b++ )
            for( int c = 0; c <= 1; c++ )
                for( int d = 0; d <= 1;d++ )
                    for( int e = 0; e <= 1; e++ )
                        for( int f = 0; f <= 1; f++ )
                        {
                                res[ 1 ][ 1 ] = a;
                                res[ 1 ][ 2 ] = b;
                                res[ 1 ][ 3 ] = c;
                                res[ 1 ][ 4 ] = d;
                                res[ 1 ][ 5 ] = e;
                                res[ 1 ][ 6 ] = f;


                                if( PressSuccessfully() )
                                {

                                    cout << "Puzzle #" << num++ << endl;
                                    for( int i = 1; i <= 5; i++ )
                                    {
                                        for( int j = 1; j <= 6; j++ )
                                        {
                                            cout << res[ i ][ j ];
                                        }
                                        cout << endl;
                                    }
                                    return 0;
                                }
                                else
                                {
                                    for( int i = 1; i <= 5; i++ )
                                        for( int j = 1; j <= 6; j++ )
                                        {
                                            press[ i ][ j ] = puzzle[ i ][ j ];
                                            memset( res, 0, sizeof( res ) );
                                        }
                                }
                        }
        return 0;
}
int main()
{

    int n;
    cin >> n;
    int i, j;

    int num = 1;

    while( n )
    {
        for( i = 1; i <= 5; i++ )
            for( j = 1; j <= 6; j++ )
            {
                cin >> puzzle[ i ][ j ];
                press[ i ][ j ] = puzzle[ i ][ j ];
                memset( res, 0, sizeof( res ) );
            }
        SelectPress( num++ );
        n--;
    }
    return 0;
}

思想：

     该问题用的是枚举法，我们知道第一排的灯泡可以由第二排的全部熄灭，第二排的可以由第3排的全部熄灭。。然后第5排可以全部熄灭第4排的。那么 最后会剩下第5行的无法进行处理。这个时候就需要枚举思想，想象一下，第一排是什么情况才会使得，进行到最后恰好第5排全部熄灭，第1排所有的情况次数是2^6次，对于这部分搜索空间。我们需要做的就是一一枚举。