熄灯问题

image.png
image.png
image.png
image.png image.png
image.png
image.png

#include<iostream>
#include<cstring>
//二进制数进行枚举

using namespace std;
char oriLights[5];//初始灯的状态
char lights[5];//中间存储的结果
char result[5];//最后的结果

int GetBit(char c, int i)
{
    return (c >> i) & 1;//右移i位与1相与，得到第i个比特
}
void SetBit(char& c, int i, int v)
{
    if (v) {
        c |= (1 << i);//c的第i位置为1
    }
    else
        c &= ~(1 << i);//c的第i位置为0；
}
void FlipBit(char& c, int i)
{
    c ^= (1 << i);//翻转
}
void OutputResult(int t, char result[])
    {
        cout << "PUZZIE #" << t << endl;
        for (int i = 0; i < 5; ++i) {
            for (int j = 0; j < 6; ++j) {
                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) {
            for (int j = 0; j < 6; ++j) {
                int s;//第i行第j列的状态
                cin >> s;
                SetBit(oriLights[i], j, s);
            }
        }
        for (int n = 0; n < 64; ++n) {//n代表第一行每个开关的状态
            int switchs = n;
            for (int i = 0; i < 5; ++i) {
                result[i] = switchs;
                memcpy(lights, oriLights, sizeof(oriLights));//oriLights复制到lights里面去
                for (int j = 0; j < 6; j++) {//第i行第j个开关
                    if (GetBit(switchs, j)) {
                        if (j > 0)
                            FlipBit(lights[i], j - 1);
                        if (j < 5)
                            FlipBit(lights[i], j + 1);

                    }

                }
                if (i < 5)
                    lights[i + 1] ^= switchs;//switchs代表第i行开关的状态；switchs第j个bit是1，则i+1行的第j个bit被翻转
                //确定i+1行的状态
                switchs = lights[i];

            }
            //最后一行的灯有没有全灭
            if (lights[4] == 0) {
                OutputResult(t, result);
                break;

            }
        }

    }
    return 0;

}