【MOOC】【枚举】4.熄灯问题——位运算

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

Input
第一行是一个正整数N，表示需要解决的案例数。每个案例由5行组成，每一行包括6个数字。这些数字以空格隔开，可以是0或1。0表示灯的初始状态是熄灭的，1表示灯的初始状态是点亮的。
Output
对每个案例，首先输出一行，输出字符串“PUZZLE #m”，其中m是该案例的序号。接着按照该案例的输入格式输出5行，其中的1表示需要把对应的按钮按下，0则表示不需要按对应的按钮。每个数字以一个空格隔开。
Sample Input
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
Sample Output
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

解题思路：
枚举第一行的开关情况，因为第一行要全部熄灭，所以第二行的开关可由第一行的开关情况以及第一行灯的状态推算出来。依次可以推算出第3，4，5行的开关，然后再验证第五行是否全熄灭。全熄灭则为正确答案。

/*
4. 熄灯问题 
-有一个由按钮组成的矩阵，其中每行有6个按钮，共5行 
-每个按钮的位置上有一盏灯
-当按下一个按钮后，该按钮以及周围位置（上下左右）的灯都会改变状态

知识点注意：
1.异或：相同位不同则为1，相同则为0 
2.将int 类型赋值给char 类型的结果
    char a;
    int b = 54;
    cout << "a : " << a << endl;
    cout << "b : " << b << endl;
    a = b;
    cout << "after a = b , a = " << a << endl;  
输出结果 a = 6;分析如下
-char a = 54;
-ASCII码54对应字符为'6'
-cout输出字符6 
*/ 
#include <iostream>
#include <cstring>
#include <string>
#include <memory.h>

using namespace std;
char oriLights[5];
char lights[5];
char result[5];

int GetBit(char c, int n)
{
    return (c >> n) & 1; 
}

void SetBit(char & c, int i, int v)
{
    if( v )
        c |= (1 << i);
    else 
        c &= ~(1 << i);        
}

void FlipBit(char & c, int i)
{
    c ^= (1 << i); 
}

void OutputResult(int t,char r[])
{
    cout << "PUZZLE #"<< t << endl;
    for(int i = 0; i < 5; ++i){
        for(int j = 0; j < 6; ++j){
            cout << GetBit(r[i],j);
            if(j < 5)
                cout << ' ';
        }
        cout << endl;
    }       
}

int main()
{
    int T;
    cin >> T;
    for(int t = 1; t <= T; ++t){
        for( int i = 0; i < 5; ++i ){
            for (int j = 0; j < 6; ++j){
                int s;
                cin >> s;
                SetBit(oriLights[i],j,s);
            }
        }
        for(int n = 0 ; n < 64 ; ++n){ //2的6次方等于64，枚举第一行从000000到111111的所有情况 
            int switchs = n;
            memcpy(lights,oriLights,sizeof(oriLights)); //函数原型:void *memcpy(void *dest, const void *src, size_t n);
            for(int l = 0; l < 5; ++l){
                result[l] = switchs;
                for(int m = 0; m < 6; ++m){
                    if( GetBit(switchs,m) ){
                        if( m > 0)
                            FlipBit(lights[l],m-1);
                        FlipBit(lights[l],m);
                        if( m < 5 )
                            FlipBit(lights[l],m+1);

                    }
                }
                if( l < 4 )
                    lights[l+1] ^= switchs; //异或：相同位不同则为1，相同则为0
                switchs = lights[l];
            }
            if(lights[4] == 0){
                OutputResult(t,result);
                break;
            }
        }
    }

    return 0;
}