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;
}
