转载自:
(68条消息) 【ACWing】95. 费解的开关_记录算法题解的博客-CSDN博客
描述
acwGPT给林克又设置了一个新关卡:“拉灯”游戏。
25盏灯排成一个5×5的方形。
每一个灯都有一个开关,游戏者可以改变它的状态。
每一步,游戏者可以改变某一个灯的状态。
游戏者改变一个灯的状态会产生连锁反应:和这个灯上下左右相邻的灯也要相应地改变其状态。
我们用数字1表示一盏开着的灯,用数字0表示关着的灯。
下面这种状态
10111
01101
10111
10000
11011
在改变了最左上角的灯的状态后将变成:
01111
11101
10111
10000
11011
再改变它正中间的灯后状态将变成:
01111
11001
11001
10100
11011
给定一些游戏的初始状态,编写程序判断游戏者是否可能在6步以内使所有的灯都变亮。
输入
第一行输入正整数n,代表数据中共有n个待解决的游戏初始状态。
以下若干行数据分为n组,每组数据有5行,每行5个字符。
每组数据描述了一个游戏的初始状态。
各组数据间用一个空行分隔。
数据范围:
0<n≤500
输出
一共输出n行数据,每行有一个小于等于6的整数,它表示对于输入数据中对应的游戏状态最少需要几步才能使所有灯变亮。
对于某一个游戏初始状态,若6步以内无法使所有灯变亮,则输出−1。
输入样例 1
3 00111 01011 10001 11010 11100 11101 11101 11110 11111 11111 01111 11111 11111 11111 11111
输出样例 1
3 2 -1
#include <iostream>
#include <cstring>
using namespace std;
const int N = 5;
char g[N][N], bg[N][N];
//灯的状态取反
void flip(char& x) {
//x=='0'的式子为真,则取‘1’,否则取‘0’。
x = x == '0' ? '1' : '0';
}
void click(int x, int y) {
//改变这个灯的状态
flip(g[x][y]);
//改变该灯上下左右四个方向灯的状态
if (x - 1 >= 0) flip(g[x - 1][y]);
if (x + 1 < 5) flip(g[x + 1][y]);
if (y - 1 >= 0) flip(g[x][y - 1]);
if (y + 1 < 5) flip(g[x][y + 1]);
}
int main() {
int T;
scanf("%d", &T);
while (T--) {
//逐行输入
for (int i = 0; i < 5; i++) scanf("%s", bg[i]);
int res = 10;
/*
对第零行每个灯枚举状态
因为前一行的灯的状态决定后一行灯的状态,
所以只需要对第零行所有灯的状态进行枚举
二进制11111-00000代表十进制31-0,
这里op=0~32表示第零行的5个灯所有可能的状态,与1取&(与),
得到的结果为:保留要枚举的元素的状态,其他元素位置清零。
当第i个位置的灯亮(为1)时,op>>i&1的值成真或为1,
这里与输入的灯的状态无关,只是枚举对第零行的处理方式,对5个灯,改变/不该变其状态。
*/
for (int op = 0; op < 32; op++) {
int cnt = 0;
memcpy(g, bg, sizeof g);//将bg矩阵中的元素复制到g,初始化。
for (int i = 0; i < 5; i++)
if (op >> i & 1) {
click(0, i);//第零行,第i列元素
cnt++;//第零行改变过状态的灯的数量
}
//第一行到第四行
for (int i = 1; i < 5; i++)
for (int j = 0; j < 5; j++)
if (g[i - 1][j] == '0') {
click(i, j);//若灯不亮,点亮灯。
cnt++;//改变过状态的灯的数量。
}
//检查第四行(最后一行)灯的状态。
bool success = true;
for (int i = 0; i < 5; i++)
if (g[4][i] == '0') {
success = false;
break;
}
//取小
if (success) res = min(res, cnt);
}
//是否所有的灯都亮过?
//若在六步以内(包括六步)就输出res,否则输出-1。
printf("%d\n", res > 6 ? -1 : res);
}
return 0;
}