原题链接
你玩过“拉灯”游戏吗?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
题解:
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
//fx为上下左右以及自身的五个方向
int n, fx[5][2] = {{-1, 0},
{1, 0},
{0, -1},
{0, 1},
{0, 0}};
char g[7][10], tmp[7][10];
//翻转第r行第c列
void change(int r, int c) {
for (int i = 0; i < 5; i++)
tmp[r + fx[i][0]][c + fx[i][1]] = '1' - tmp[r + fx[i][0]][c + fx[i][1]] + '0';
}
void dfs(int k, int &cnt) {
int count = 0;
//枚举第一行
for (int j = 0; j < 5; j++)
if (k & (1 << j)) {
++count;
change(1, j + 1);
}
//第2-4行
for (int i = 2; i <= 5; i++) {
for (int j = 1; j <= 5; j++) {
if (tmp[i - 1][j] == '0') {
++count;
change(i, j);
}
}
}
for (int i = 1; i <= 5; i++)
if (tmp[5][i] == '0')
return;
cnt = min(cnt, count);
}
int main() {
cin >> n;
while (n--) {
//初值大于6即可
int cnt = 10;
//从第1行开始存,这样不需要越界判断
for (int i = 1; i <= 5; i++)
scanf("%s", &g[i][1]);
//利用位运算枚举第一行可能的翻转情况,这样在判断2-5行时,必须把前一行为0的方块反转为1
for (int i = 0; i < (1 << 5); i++) {
for (int j = 1; j <= 5; j++)
memcpy(tmp[j], g[j], 8);
dfs(i, cnt);
}
if (cnt <= 6)
cout << cnt << endl;
else
cout << "-1" << endl;
}
return 0;
}