熄灯问题(枚举+状态压缩）

最新推荐文章于 2023-12-20 22:08:33 发布

·马克图布·

最新推荐文章于 2023-12-20 22:08:33 发布

阅读量465

点赞数 1

分类专栏：状态压缩

本文链接：https://blog.csdn.net/qq_43408238/article/details/89569668

版权

状态压缩专栏收录该内容

13 篇文章 0 订阅

订阅专栏

描述

有一个由按钮组成的矩阵，其中每行有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列的灯。

输入

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

输出

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

样例输入

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

样例输出

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

因要求灯全部熄灭，只需要枚举第一行，以局部确定整体，可大大减少枚举的范围。又因为是5*6的矩阵，所以可用字符数组保存灯的状态，一个bit代表着一盏灯，通过位运算很方便的改变灯的状态。

AC Code：

#include <iostream>
#include<string>
#include<cstring>
#include<memory>
using namespace std;
char oriLights[5];//最初灯矩阵，一个bite表示一盏灯
char lights[5];//不停变换的灯矩阵
char result[5];//结束开关矩阵
int GetBit(char c,int i)//取c的第i位
{
    return (c>>i)&1;
}
void SetBit(char & c,int i,int v)//设置c的第i位为v
{
    if(v){
        c|=(1<<i);
    }
    else
        c &= ~(1<<i);
}
void FlipBit(char & c,int i)//将c的第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;++j){
            cout << GetBit(result[i],j);
            if( j<5)
                cout <<" ";
        }
        cout << endl;
    }
}
int main()
{
    int T;
    //cin >> T;
    T=1;
    for(int t=1;t<=T;++t){//读入灯的最初状态
        for(int i=0;i<5;++i)
            for(int j=0;j<6;++j){
                int s;
                cin>>s;
                SetBit(oriLights[i],j,s);
        }
        for(int n=0;n<64;++n){//遍历首行开关的64种状态
            int switchs = n;//第i行的开关状态
            memcpy(lights,oriLights,sizeof(oriLights));
            for(int i=0;i<5;++i){
                result[i] = switchs;//第i行的开关方案
                for(int j=0;j<6;j++){
                    if(GetBit(switchs,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] ^= switchs;//改下一行的灯
                switchs = lights[i];//第i+1行开关方案和第i行灯情况相同
            }
            if(lights[4] == 0){
                OutputResult(t,result);
                break;
            }
        }
    }
    return 0;
}