枚举之熄灯问题

题目：有一个由按钮组成的矩阵，其中每行有6个按钮，共5行。每个按钮的位置上有一盏灯。当按下一个按钮后，该按钮以及周围位置(上边、下边、左边、右边)的灯都会改变一次。即，如果灯原来是点亮的，就会被熄灭；如果灯原来是熄灭的，则会被点亮。在矩阵角上的按钮改变3盏灯的状态；在矩阵边上的按钮改变4盏灯的状态；其他的按钮改变5盏灯的状态。请你写一个程序，确定需要按下哪些按钮，恰好使得所有的灯都熄灭。

#include<iostream>
#include<string>
#include<cstring>
#include<memory>
using namespace std;
char oriLights[5];
char lights[5];
char result[5];

int GetBit(char c, int i)
{
    return (c >> i) & 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 result[])
{
    cout << "PUZZLE #" << t << endl;
    for(int i=0;i<5;++i)
        for (int j = 0; j < 6; ++i) {
            cout << GetBit(result[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)//original input 43-48 input one line
            for (int j = 0; j < 6; ++j) {
                int s;
                cin >> s;
                SetBit(oriLights[i], j, s);
            }
        for (int n = 0; n < 64; ++n) {
            int switches = n;
            memcpy(lights, oriLights, sizeof(oriLights));
            for (int i = 0; i < 5; ++i) {
                result[i] = switches;
                for (int j = 0; j < 6; j++) {
                    if (GetBit(switches, j)) {
                        if (j > 0)
                            FlipBit(lights[i], j - 1);
                        FlipBit(lights[i], j);
                        if (j < 5)
                            FlipBit(lights[i], j + 1);
                    }
                }
                if (i < 4)
                    lights[i + 1] ^= switches;
                switches = lights[i];
            }
            if (lights[4] == 0) {
                OutPutResult(t, result);
                break;
            }
        }
    }
}