题目链接:AcWing 95. 费解的开关
你玩过“拉灯”游戏吗?25盏灯排成一个5x5的方形。每一个灯都有一个开关,游戏者可以改变它的状态。每一步,游戏者可以改变某一个灯的状态。游戏者改变一个灯的状态会产生连锁反应:和这个灯上下左右相邻的灯也要相应地改变其状态。
我们用数字“1”表示一盏开着的灯,用数字“0”表示关着的灯。下面这种状态
10111
01101
10111
10000
11011
在改变了最左上角的灯的状态后将变成:
01111
11101
10111
10000
11011
再改变它正中间的灯后状态将变成:
01111
11001
11001
10100
11011
给定一些游戏的初始状态,编写程序判断游戏者是否可能在6步以内使所有的灯都变亮。
输入格式
第一行输入正整数n,代表数据中共有n个待解决的游戏初始状态。
以下若干行数据分为n组,每组数据有5行,每行5个字符。每组数据描述了一个游戏的初始状态。各组数据间用一个空行分隔。
输出格式
一共输出n行数据,每行有一个小于等于6的整数,它表示对于输入数据中对应的游戏状态最少需要几步才能使所有灯变亮。
对于某一个游戏初始状态,若6步以内无法使所有灯变亮,则输出“-1”。
数据范围
0<n≤500
输入样例:
3
00111
01011
10001
11010
11100
11101
11101
11110
11111
11111
01111
11111
11111
11111
11111
输出样例:
3
2
-1
程序说明:
第一行有5个开关,每个开关有两种状态:按下和不按,一共有2^5种情况。可以用0~31的二进制来表示每一位的状态:1表示按下,0表示不按下,如10001表示第一个和最后一个开关按下。
32个二进制数对应第一行的32种情况,每种情况只看前4行。如果某一位是关着的,那么需要将它下一行所对应的开关按下,即每个开关的状态必须被下一行对应的开关改变。
循环最后一行,如果所有的开关都打开则有效,否则无效。
代码如下:
#include <iostream>
#include <cstring>
using namespace std;
char g[6][6];
int n, dx[] = {0, 0, 0, 1, -1}, dy[] = {0, 1, -1, 0, 0};
void turn(int x, int y) {
for(int i = 0; i < 5; i++) {
int x1 = x + dx[i], y1 = y + dy[i];
if(x1 >= 0 && y1 >= 0 && x1 < 5 && y1 < 5) {
g[x1][y1] ^= 1;
}
}
}
int main() {
cin>>n;
while(n--) {
for(int i = 0; i < 5; i++)
for(int j = 0; j < 5; j++)
cin>>g[i][j];
//每次循环都会改变g[][]的内容,因此需要先备份一下
char backup[6][6];
memcpy(backup, g, sizeof g);
int res = 1e9;
//枚举第一行的32种情况
for(int i = 0; i < 32; i++) {
int cnt = 0;
for(int j = 0; j < 5; j++) {
if(i >> j & 1) {
turn(0, j);
cnt++;
}
}
//枚举前4行
for(int j = 0; j < 4; j++) {
//枚举每一行的5个位置
for(int k = 0; k < 5; k++) { //枚举每一行的5个开关
if(g[j][k] == '0') {
turn(j + 1, k); //按下下一行的开关
cnt++;
}
}
}
//枚举最后一行
bool flag = true;
for(int j = 0; j < 5; j++) {
if(g[4][j] == '0') {
flag = false;
break;
}
}
if(flag) res = min(res, cnt); //记录每种方案的最小值
memcpy(g, backup, sizeof backup);
}
if(res > 6) cout<<-1<<endl;
else cout<<res<<endl;
}
return 0;
}