算法竞赛进阶指南打卡系列题解
题目
原题链接
题面
你玩过“拉灯”游戏吗?
25 25 25 盏灯排成一个 5 × 5 5×5 5×5 的方形。
每一个灯都有一个开关,游戏者可以改变它的状态。
每一步,游戏者可以改变某一个灯的状态。
游戏者改变一个灯的状态会产生连锁反应:和这个灯上下左右相邻的灯也要相应地改变其状态。
我们用数字 1 1 1 表示一盏开着的灯,用数字 0 0 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
0<n≤500
0<n≤500
输入样例:
3
00111
01011
10001
11010
11100
11101
11101
11110
11111
11111
01111
11111
11111
11111
11111
输出样例:
3
2
-1
题解
思路
首先看一眼数范围,再看一眼标题,暴力大法好。(doge)
首先清楚一个性质,重复点亮一个灯是没有意义的,因此我们可以一行一行做,每次将一行的所有灯点亮,即可避免重复枚举。
由于第 i i i 行灯的亮灭是由第 i + 1 i+1 i+1 行控制的,因此假设此时我们想要将第 ( i , j ) (i,j) (i,j) 个灯由 0 变为 1 ,只需要改变 ( i + 1 , j ) (i +1, j) (i+1,j) 状态即可。(因为只有这样才能不影响到第 i i i 行之前的灯的状态)
判断是否合理,我们可以把上述思路从第 1 1 1 行到第 4 4 4 行实现,然后判断第 5 5 5 行的状态是否合法即可。
仅此而已肯定是不够的,因为由于第一行状态已经确定,按照上面的思路做的话只有一种解,而且这个解是从第二行开始改变灯的状态而得到的,显然不完备,因为有一些解显然是需要改变第一行的灯才能做到的。
因此我们可以用二进制枚举第一行的状态( 1 1 1 为改变, 0 0 0 为不改变,无论亮灭),在按照上述思路改变第 1 1 1 至 4 4 4 行的状态,最后判断即可。
第一行的状态一共 2 5 − 1 2^5-1 25−1 种状态,很少。
代码
#include <bits/stdc++.h>
// #define int long long
using namespace std;
typedef long long LL;
void solve()
{
vector<string> a(5);
for (int i = 0; i < 5; i ++ ) cin >> a[i];
auto ba = a; // 备份一下a数组方便枚举第一行状态
int dx[] = {-1, 0, 1, 0, 0}, dy[] = {0, 1, 0, -1, 0};
function<void(int, int)> turn = [&](int x, int y) { // 写个函数实现改变灯的状态
for (int i = 0; i < 5; i ++ ) {
int a = x + dx[i], b = y + dy[i];
if (a < 0 || a >= 5 || b < 0 || b >= 5) continue;
ba[a][b] ^= 1;
}
};
int ans = 10;
for (int i = 0; i < 32; i ++ ) { // 一共32个状态
int step = 0;
ba = a;
for (int j = 0; j < 5; j ++ ) { // 直接根据二进制数改变第一行的状态,无论亮灭
if (i >> j & 1) {
turn(0, j);
step ++ ;
}
}
for (int j = 0; j < 4; j ++ ) { // 改变状态
for (int k = 0; k < 5; k ++ ) {
if (ba[j][k] == '0') {
turn(j + 1, k);
step ++ ;
}
}
}
bool flag = true; // 判断是否合法
for (int j = 0; j < 5; j ++ ) {
if (ba[4][j] == '0') {
flag = false;
break;
}
}
if (flag) ans = min(ans, step);
}
cout << (ans > 6 ? -1 : ans) << "\n";
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int T;
cin >> T;
while (T -- ) {
solve();
}
return 0;
}