蓝桥杯 ADV-140 算法提高 开灯游戏

算法提高 开灯游戏  

时间限制：1.0s   内存限制：256.0MB

 

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

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

　　具体如下：

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

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

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

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

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

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

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

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

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

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

 

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

 

样例输出
000001011
000001110
000001111
 

#include <stdio.h>

int light_on(int switches)
{
    int lights[9] = { 0 };

    if (switches & 1)
    {
        lights[5] = !lights[5];
        lights[7] = !lights[7];
    }
    switches >>= 1;
    if (switches & 1)
    {
        lights[4] = !lights[4];
        lights[6] = !lights[6];
        lights[8] = !lights[8];
    }
    switches >>= 1;
    if (switches & 1)
    {
        lights[3] = !lights[3];
        lights[7] = !lights[7];
    }
    switches >>= 1;
    if (switches & 1)
    {
        lights[2] = !lights[2];
        lights[4] = !lights[4];
        lights[8] = !lights[8];
    }
    switches >>= 1;
    if (switches & 1)
    {
        lights[1] = !lights[1];
        lights[3] = !lights[3];
        lights[5] = !lights[5];
        lights[7] = !lights[7];
    }
    switches >>= 1;
    if (switches & 1)
    {
        lights[0] = !lights[0];
        lights[4] = !lights[4];
        lights[6] = !lights[6];
    }
    switches >>= 1;
    if (switches & 1)
    {
        lights[1] = !lights[1];
        lights[5] = !lights[5];
    }
    switches >>= 1;
    if (switches & 1)
    {
        lights[0] = !lights[0];
        lights[2] = !lights[2];
        lights[4] = !lights[4];
    }
    switches >>= 1;
    if (switches & 1)
    {
        lights[1] = !lights[1];
        lights[3] = !lights[3];
    }

    int num_lights_on = 0;
    for (int i = 0; i < 9; ++i)
        if (lights[i])
            num_lights_on++;
    return num_lights_on;
}

void print_switch(int switches)
{
    int bits[9] = { 0 };
    for (int i = 8; i >= 0; --i)
    {
        bits[i] = switches & 1;
        switches >>= 1;
    }
    for (int i = 0; i < 9; ++i)
        printf("%d", bits[i]);
    printf("\n");
}

int main()
{
    for (int switches = 0; switches < 1 << 9; ++switches)
    {
        if (light_on(switches) == 4)
            print_switch(switches);
    }

    return 0;
}