2811:熄灯问题

描述

有一个由按钮组成的矩阵，其中每行有6个按钮，共5行。每个按钮的位置上有一盏灯。当按下一个按钮后，该按钮以及周围位置(上边、下边、左边、右边)的灯都会改变一次。即，如果灯原来是点亮的，就会被熄灭；如果灯原来是熄灭的，则会被点亮。在矩阵角上的按钮改变3盏灯的状态；在矩阵边上的按钮改变4盏灯的状态；其他的按钮改变5盏灯的状态。
请你写一个程序，确定需要按下哪些按钮，恰好使得所有的灯都熄灭。根据上面的规则，我们知道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则表示不需要按对应的按钮。

思路

假设第i行的第j列需要改变状态，而第i行的其他不需要改变，则可以通过改变i+1行或者i-1行的第j列来改变(i, j)，通过这个规律可以知道当我对第1行进行某种改变后，整个矩阵的最终值就已经确定了，所以只要遍历第1行的2^64-1次就可以了。

代码

#include<bits/stdc++.h>
using namespace std;

int arr[10][10];//初始数组
int init[10][10];//按钮

bool check()//检查是否符合要求
{
    int res[10][10] = {0};//存入当前init情况下的结果
    for(int i = 1; i <= 5; i++)
    {
        for(int j = 1; j <= 6; j++)
        {
            res[i][j] = (arr[i][j] + init[i][j] + init[i-1][j] + init[i][j-1] + init[i][j+1])%2;
            init[i+1][j] = res[i][j];
            //将最终状态存入res数组
        }
    }
    for(int i = 1; i <= 6; i++)
    {
        if(res[5][i] == 1)
        {
            return false;
            //由于前面几行一定是完成了的，所以只需要判断最后一行是否符合条件
        }
    }
    return true;
}

void change()//枚举各种情况并判断合理性
{
    for(int i = 0; i <= 6; i++)
    {
        init[1][i] = 0;
    }
    //对第一行初始化
    while(check() == false)//改变init数组的第一行
    {
        init[1][1]++;
        int i = 1;
        while(init[1][i] > 1)
        {
            init[1][i] = 0;
            i++;
            init[1][i]++;
        }//枚举每一种情况(第一次看的时候没看懂，可以理解成高精度二进制)
    }
    return;
}

int main()
{
    for(int i = 1; i <= 5; i++)
    {
        for(int j = 1; j <= 6; j++)
        {
            cin >> arr[i][j];
        }
    }
    for(int i  = 0; i < 7; i++)
    {
        arr[i][0] = 0;
        arr[0][i] = 0;
    }
    
    change();
    
    for(int i = 1; i <= 5; i++)
    {
        for(int j = 1; j <= 6; j++)
        {
            cout << init[i][j] << ' ';
        }
        cout << endl;
    }
    return 0;
}