熄灯问题（百练2811）

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

/*
解题思路：
给定房间灯光现状矩阵matrix，令操作矩阵press全取0；
不管第一行的灯光矩阵是怎么样的，第二行的press都有办法把第一行的灯光全部熄灭；
依次类推，第i行的press总有办法把第i-1行的灯光全部熄灭；
因此，只要第一行的press取值定下来，一定有办法让第1到第4行的灯光全部熄灭；
然后考虑第5行，此时已经没有办法通过press改变第5行的灯光；
因此，第5行灯光在第5行的使得第4行灯光全部熄灭的press操作下，必须恰好全部熄灭；
因此思路是，枚举第一行的press，找出恰好实现第5行全部熄灭的那个press；
*/

int matrix[6][8];
int press[6][8];
int caseNum;

bool guess(){//通过一个给定的press矩阵第一行的设置，考察能否得出解
    /*
    考虑第2行至第5行的press；
    第2行的press用于让第1行的matrix矩阵中的值全变为0；
    第i行第j列的press值取决于以下因素：
    目标房间的初始状态matrix[i-1][j];
    四个对房间状态进行改变的press操作，分别为：
    press[i-2][j],press[i-1][j-1],press[i-1][j],press[i-1][j+1]；
    如果初始状态1，press加总%2为1，则应取press为0；
    如果初始状态1，press加总%2为0，则应取press为1；
    如果初始状态0，press加总%2为1，则应取press为1；
    如果初始状态0，press加总%2为0，则应取press为0；
    总结，press应取为(初始状态+press加总%2)%2；
    */
    for(int i=2;i<=5;i++){
        for(int j=1;j<=6;j++){
            press[i][j]=(matrix[i-1][j]+press[i-2][j]+press[i-1][j-1]+press[i-1][j]+press[i-1][j+1])%2;
        }
    }
    /*
    第5行的press操作使得第4行的matrix状态值全部为0；
    判断此时第5行的matrix值是否为0；
    由于之前的操作不对matrix矩阵进行修改；
    因此直接判断press操作加总是否使得对应的matrix值取为0；
    如果初始状态1，press加总%2为1，则恢复为0；
    如果初始状态0，press加总%2为0，则恢复为0；
    否则为1；
    总结，若(初始状态+press加总%2)%2为0，则成功熄灭；
    否则未成功熄灭；
    */
    for(int i=1;i<=6;i++){
        if((matrix[5][i]+press[5][i-1]+press[5][i]+press[5][i+1]+press[4][i])%2)
            return false;
    }
    return true;

}

void enumerate(){
    for(int i=1;i<=6;i++){//把press的第一行初始化为000000
        press[1][i]=0;
    }
    while(guess()==false){//如果这样的第一行设置不能得出解，注意用while循环
        press[1][1]++;
        int c=1;//考虑最后一位开始的进位
        while(press[1][c]>1){
            press[1][c]=0;//进位，原位恢复为0
            c++;
            press[1][c]++;
        }
    }
    //否则，第一行的设置能够得出解
    return;

}

int main()
{
    cin>>caseNum;
    int CaseCount = 1;
    while(caseNum--){
        memset(matrix,0,sizeof(matrix));
        memset(press,0,sizeof(press));
        for(int i=1;i<=5;i++){//输入
            for(int j=1;j<=6;j++){
                cin>>matrix[i][j];
            }
        }
        enumerate();//运算
        cout<<"PUZZLE #"<<CaseCount<<endl;
        CaseCount++;
        for(int i=1;i<=5;i++){//输出
            for(int j=1;j<=6;j++){
                cout<<press[i][j]<<" ";
            }
            cout<<endl;
        }
    }
    return 0;
}