1520 问题 D: 蓝桥杯算法提高VIP-开灯游戏

题目描述

有9盏灯与9个开关，编号都是1~9。

每个开关能控制若干盏灯，按下一次会改变其控制的灯的状态(亮的变成不亮，不亮变成亮的)。

具体如下：

第一个开关控制第二，第四盏灯；

第二个开关控制第一，第三，第五盏灯；

第三个开关控制第二，第六盏灯；

第四个开关控制第一，第五，第七盏灯；

第五个开关控制第二，第四，第六，第八盏灯；

第六个开关控制第三，第五，第九盏灯；

第七个开关控制第四，第八盏灯；

第八个开关控制第五，第七，第九盏灯；

第九个开关控制第六，第八盏灯。

开始时所有灯都是熄灭的，开关是关闭着的。要求按下若干开关后，使得只有4盏灯亮着。

输出

输出所有可能的方案，每行一个方案，每一行有9个字符，从左往右第i个字符表示第i个开关的状态(" 0" 表示关闭，" 1" 表示打开)，按字典序输出。下面的样例输出只是部分方案。

解答

#include <iostream>

using namespace std;
int one = 2 + 8;
int two = 1 + 4 + 16;
int three = 2 + 32;
int four = 1 + 16 + 64;
int five = 2 + 8 + 32 + 128;
int six = 4 + 16 + 256;
int seven = 8 + 128;
int eight = 16 + 64 + 256;
int nine = 32 + 128;
int op[9] = {one, two, three, four, five, six, seven, eight, nine};

int main()
{
    int n = 0;
    for (int i = 0; i <= 512; i++)
    {
        n = 0;
        for (int j = 0; j < 9; j++)
        {
            if (i & (1 << j))
            {
                n ^= op[j];
            }
        }
        int ans = 0;
        for (int j = 0; j < 9; j++)
        {
            if (n & (1 << j))
            {
                ans++;
            }
        }
        if (ans == 4)
        {
            for (int j = 8; j >= 0; j--)
            {
                if (i & (1 << j))
                {
                    cout << '1';
                }
                else
                {
                    cout << '0';
                }
            }
            cout << endl;
        }
    }
    return 0;
}