百练：2811:熄灯问题

#include<bits/stdc++.h> 
using namespace std;
int n = 5, m = 6;
//a是初始灯的状态，b是按钮的按压状态 
int a[5][6], b[5][6], temp[5][6]; 
int d[][2] = {{-1, 0}, {0, 1}, {1, 0}, {0, -1}};//四个方向 
//检查下标是否越界
bool check(int x, int y)
{ return x >= 0 && x < n && y >= 0 && y < m; }
     //灯的状态翻转
void tot(int x, int y){ temp[x][y] = !temp[x][y]; } 
      //按压按钮 
 void press(int x, int y){
   //自身的状态翻转 
       tot(x, y); 
  //记录按压 
  b[x][y] = 1;
   //影响四个方向
         for(int k = 0; k < 4; k++)
         { int nx = x + d[k][0], ny = y + d[k][1];
          if(check(nx, ny)) tot(nx, ny); } } 
 //判断当前是否还有亮灯
bool isOk()
           { 
           for(int i = 0; i < n; i++) 
           for(int j = 0; j < m; j++)
            if(temp[i][j]) return false; return true; } 
int main(){    
    for(int i = 0; i < n; i++)
    for(int j = 0; j < m; j++) 
             scanf("%d", &a[i][j]); //枚举第一行的按钮状态
     for(int i = 0; i < (1 << m); i++){ 
              memcpy(temp, a, sizeof a);
         //每次初始时，没有按钮被按，故初始为0
            memset(b, 0, sizeof b); 
        //先把第一行需要按的按钮按了
            for(int j = 0; j < m; j++) 
                 if(i >> j & 1)
                 press(0, j); 
 //枚举后面的每一行，根据上面一行的状态，决定自身是否需要按    
                 for(int k = 1; k < n; k++) 
                 for(int j = 0; j < m; j++) 
                 if(temp[k - 1][j]) 
                 press(k, j); 
                 //所有按钮都枚举完了，判断当前按法是否符和要求 
                 if(isOk()) break; } 
                 //输出 
                 for(int i = 0; i < n; i++)
                 { for(int j = 0; j < m; j++) 
                 printf("%d ", b[i][j]);
                 puts("");
                  } 
                  return 0;
                   }
总时间限制: 1000ms 内存限制: 65536kB
描述
有一个由按钮组成的矩阵，其中每行有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
2811:熄灯问题