OpenJudege-2811 熄灯问题 解题报告

OpenJudge - 2811

问题描述

熄灯问题

有一个由按钮组成的矩阵，其中每行有6个按钮，共5行。每个按钮的位置上有一盏灯。当按下一个按钮后，该按钮以及周围位置(上边、下边、左边、右边)的灯都会改变一次。即，如果灯原来是点亮的，就会被熄灭；如果灯原来是熄灭的，则会被点亮。在矩阵角上的按钮改变3盏灯的状态；在矩阵边上的按钮改变4盏灯的状态；其他的按钮改变5盏灯的状态。
在上图中，左边矩阵中用X标记的按钮表示被按下，右边的矩阵表示灯状态的改变。对矩阵中的每盏灯设置一个初始状态。请你按按钮，直至每一盏等都熄灭。与一盏灯毗邻的多个按钮被按下时，一个操作会抵消另一次操作的结果。在下图中，第2行第3、5列的按钮都被按下，因此第2行、第4列的灯的状态就不改变。
请你写一个程序，确定需要按下哪些按钮，恰好使得所有的灯都熄灭。根据上面的规则，我们知道1）第2次按下同一个按钮时，将抵消第1次按下时所产生的结果。因此，每个按钮最多只需要按下一次；2）各个按钮被按下的顺序对最终的结果没有影响；3）对第1行中每盏点亮的灯，按下第2行对应的按钮，就可以熄灭第1行的全部灯。如此重复下去，可以熄灭第1、2、3、4行的全部灯。同样，按下第1、2、3、4、5列的按钮，可以熄灭前5列的灯。

Input

5行组成，每一行包括6个数字（0或1）。相邻两个数字之间用单个空格隔开。0表示灯的初始状态是熄灭的，1表示灯的初始状态是点亮的。

Output

5行组成，每一行包括6个数字（0或1）。相邻两个数字之间用单个空格隔开。其中的1表示需要把对应的按钮按下，0则表示不需要按对应的按钮。

Sample Input

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

Sample Output

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

题目大意

熄灯问题，顾名思义，要将给定的灯矩阵全部熄灭，其中，灯亮用1表示，灯灭用0表示，即最终要使灯矩阵变为0矩阵。输出中，1表示按下相应灯按钮，0表示不按相应灯按钮，最终要求输出一个按按钮的方案，能使得所有灯熄灭。
另外还有一点要注意，每个灯最多只需要按一次，不存在一盏灯按多次的状况，因为两次按下一盏灯，相当于没按。

解题方法

首先这是一道枚举。

---

首先是枚举方法

但是要枚举，我们不能枚举所有的灯的方案，这样共有$2^{30}$种，会超时，因此要设计一种较少的枚举方法。
我们有以下思路：
枚举第一行的所有状态，在第二行，按下第一行没有熄灭的灯对应的第二行按钮，这样，第一行灯即可完全熄灭。如：

(按下第二行的几盏灯，使得第一行的所有灯熄灭)
第二行、第三行、第四行、第五行以此类推，这样我们一定能够保证前四行的灯全部熄灭，再判断第五行灯的状况，若全部熄灭，则说明这种方法可行，因为题目只要求输出一种方案，则输入这种方案，程序结束。若第五行的灯没有全部熄灭，说明第一行的枚举方式不正确，需要重新枚举。
这样，我们就将枚举整个矩形的工程简化为枚举第一行的工程，此时最多需要枚举$2^6$，即64种方案。

---

接下来是灯和状态的表示方法。
要枚举和表示这些01组合，用一个十进制数组，再进行6重循环的方法可行，但浪费空间，而且不够简洁。因此我们可以设想用二进制数字枚举，而一个char类型的字符占一个字节，8个比特，这已经足够我们枚举所有的方案，因此，只需要使用char类型中ASCII码为0的字符到ASCII码为63的字符，便可以枚举到所有组合。(二进制从右往左数以此为第1位、第2位…，在枚举中，我们用第 i(1 <= i <= 6) 位模拟第 i 盏灯的状况。)
在按按钮的方案中，我们同样适用这种二进制表示法，在输出时，只需要通过位运算，输出每个每个二进制位即可。
下列函数为几个需要用到的二进制操作函数。

---

获取二进制的某一位

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

设置二进制的某一位

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

翻转二进制的某一位

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

---

最后是确定某一行灯的状况
我们可以通过二进制异或操作方便的获取某一行灯在按下按钮后的状况
即 1^1 = 0，一个亮灯按一次变为0(灭)，1^0 = 1，一个灯不按还是1(亮)，0^0 = 0，一个灭灯不按为0(灭)，0^1 = 1，一个灭灯按一次变为1(亮)。
因此，通过一个异或运算就可以获得当前行的灯的状况。

代码如下

#include <iostream>
#include <cstdio>
#include <cstring>
#include <memory>
using namespace std;
int T;
char oriLights[10];
char Lights[10];
char result[10];
int GetBit(char c, int i)
{
    return (c >> i) & 1;
}
void SetBit(char &c, int i, int v)
{
    if(v)
        c = c | (1 << i);
    else
        c = c & ~(1 << i);
}
void FilpBit(char &c, int i)
{
    c = c ^ (1 << i);
}
void OutputResult(int t)
{
    cout << "PUZZLE #" << 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 << " ";
            else
                cout << endl;
        }
}
int main()
{
    cin >> T;
    char switches;
    for(int t = 1; t <= T; t ++)
    {
        memset(oriLights, 0, sizeof(oriLights));
        int v;
        for(int i = 0; i < 5; i ++)
            for(int j = 0; j < 6; j ++)
            {
                cin >> v;
                SetBit(oriLights[i], j, v);
            }
        for(char c = 0; c < 64; c ++)
        {
            memcpy(Lights, oriLights, sizeof(oriLights));
            switches = c;
            for(int i = 0; i < 5; i ++)
            {
                result[i] = switches;
                for(int j = 0; j < 6; j ++)
                    if(GetBit(switches, j))
                    {
                        if(j > 0)
                            FilpBit(Lights[i], j - 1);
                        FilpBit(Lights[i], j);
                        if(j < 5)
                            FilpBit(Lights[i], j + 1);
                    }
                if(i < 4)
                    Lights[i + 1] ^= switches;
                switches = Lights[i];
            }
            if(Lights[4] == 0)
            {
                OutputResult(t);
                break;
            }
        }
    }
    return 0;
}