题目:有一个由按钮组成的矩阵,其中每行有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;
}
}
}
}