bttcOj1081 灯光师

Description

Jeremy是师范学院学生艺术团的灯光师，今晚的的艺术表演的灯光由他来控制，控制舞台的灯是由一些开关组成的矩阵，矩阵的每行有6个开关，总共有5行。每个开关位置对应一盏灯的位置。但是并不是每个开关都只能控制一盏灯。当按下一个开关之后，该开关对应位置以及其周围位置(上、下、左、右)的灯都会改变一次状态。如果灯原来是点亮的，就会被熄灭；如果灯原来是熄灭的，则会被点亮。显然在矩阵角上的开关改变 3 盏灯的状态 ，在矩阵边上的按钮改变 4 盏灯的状态，其他的开关改变 5 盏灯的状态。现在需要你帮助Jeremy判断需要按下哪些按钮，恰好使得所有的灯都熄灭。

Input

多组输入数据，每组第一行是一个正整数N，表示需要解决的案例数。每个案例由5行组成，每一行包括6个数字。

这些数字以空格隔开，0 或 1 。 0表示灯的初始状态是熄灭的，1表示灯的初始状态是点亮的。

Output

对每个案例，首先输出一行，输出字符串“PUZZLE #m”，其中m是该案例的序号。接着按照该案例的输入格式输出5行，其中的 1 表示需要把对应的按钮按下，0 则表示不需要按对应的按钮。每个数字以一个空格隔开。

Sample Input

2
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
0 0 1 0 1 0
1 0 1 0 1 1
0 0 1 0 1 1
1 0 1 1 0 0
0 1 0 1 0 0

Sample Output

PUZZLE #1
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
PUZZLE #2
1 0 0 1 1 1
1 1 0 0 0 0
0 0 0 1 0 0
1 1 0 1 0 1
1 0 1 1 0 1 

思路：枚举第一行所有按灯情况，每次从第二行开始遍历，看当前位置头顶的灯是否亮，如果亮，将当前位置按下，且将把周围影响的灯都对应上。遍历完整个数组后，看最后一行是否全为灭，如果全为免即所有灯已灭，否则，继续枚举第一行的按灯情况，直到全部按灭位置。

因为每盏灯只有俩种情况，所以第一行一种2^6种情况！

参考：敏就是NB，这是他想出来的！

代码：

#include <iostream>
#include <string.h>
using namespace std;
int visit[5][6];//记录按灯位置
void status(int i,int j,int a[][6])//按灯后周围灯的影响
{
    a[i][j] = !a[i][j];
    if(i>0) a[i-1][j] = !a[i-1][j];
    if(i<4) a[i+1][j] = !a[i+1][j];
    if(j>0) a[i][j-1] = !a[i][j-1];
    if(j<5) a[i][j+1] = !a[i][j+1];
}
bool fun(int orig[][6])
{
    int a[5][6];
    for(int i=0;i<5;i++)
        for(int j=0;j<6;j++) a[i][j] = orig[i][j];
    for(int j=0;j<6;j++) if(visit[0][j]) status(0,j,a);//枚举第一行按灯
    for(int i=1;i<5;i++)
    {
        for(int j=0;j<6;j++)
        {
            if(a[i-1][j] == 1)//每次灭头顶上的灯
            {
                status(i,j,a);
                visit[i][j] = 1;
            }
        }
    }
    int flag = 1;
    for(int j=0;j<6;j++)//查看最后一行是否为全灭状态
        if(a[4][j]) {flag = 0;break;}
    if(flag)return true;
return false;}
void translate(int num)//转换二进制
{
    for(int i=5;i>=0;i--)
    {
        visit[0][i] = num % 2;
        num /= 2;
    }
}
int main()
{
    int n,a[5][6];
    while(scanf("%d",&n)!=EOF)
    {
        for(int c=1;c<=n;c++)
        {
            for(int i=0;i<5;i++)
                for(int j=0;j<6;j++) scanf("%d",&a[i][j]);
            for(int i=0;i<64;i++)//枚举第一行按灯情况，每盏灯只有俩种情况，一种2的6次方种情况
            {
                memset(visit,0,sizeof(visit));
                translate(i);
                if(fun(a)) break;//全灭
            }
            printf("PUZZLE #%d\n",c);
            for(int i=0;i<5;i++)
            {
                for(int j=0;j<5;j++) printf("%d ",visit[i][j]);
                printf("%d\n",visit[i][5]);
            }
        }
    }
    return 0;
}