递推 费解的开关 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

代码：

#include <iostream>
#include <cstring>

using namespace std;
const int INF = 1000000;
char g[10][10];
int dx[5] = {0, -1, 0, 1, 0};
int dy[5] = {0, 0, 1, 0, -1};

void 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)
            g[a][b] ^= 1;
    }
}

int work() {
    int ans = INF;
    for (int i = 0; i < 1 << 5; ++i) {
        int res = 0;
        char backup[10][10];
        memcpy(backup, g, sizeof g);
        //先遍历第一行的所有情况（共有 2^5 三十二种，
        //故外层循环要循环32次来枚举第一行的所有操作）
        for (int j = 0; j < 5; ++j) {
            if (i >> j & 1) {
                res++;
                turn(0, j);
            }
        }
        //从第二行开始递推，算出第一行该操作所对应的唯一res
        for (int j = 0; j < 4; ++j) {
            for (int k = 0; k < 5; ++k) {
                if (g[j][k] == '0') {
                    res++;
                    turn(j + 1, k);
                }
            }
        }
        //判断将灯全部打开
        bool isSucc = true;
        for (int j = 0; j < 5; ++j) {
            if (g[4][j] == '0') {
                isSucc = false;
                break;
            }
        }
        if (isSucc)ans = min(ans, res);
        memcpy(g, backup, sizeof g);
    }
    if (ans > 6)return -1;
    return ans;
}

int main() {
    int n;
    cin >> n;
    while (n--) {
        for (int i = 0; i < 5; ++i)
            cin >> g[i];
        cout << work() << endl;
    }
    return 0;
}