【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\le 500$

输入样例

3
00111
01011
10001
11010
11100

11101
11101
11110
11111
11111

01111
11111
11111
11111
11111

输出样例

3
2
-1

算法

这道题结合了递归与递推的思想，首先需要去枚举第一行的开关操作状态（可能是操作1，2，3，4，5个），所以不能使用两重循环。递推的话，就是根据前一行灯的状态，来改变当前行的状态。

#include <bits/stdc++.h>

using namespace std;
const int N = 10;
int arr[N];
char mat[N][N], back[N][N];
int res;
int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};
void turn(int x, int y) { // 改变mat[x][y]，以及周为的状态
    mat[x][y] = mat[x][y] == '0' ? '1' : '0';
    for (int i = 0; i < 4; ++ i) {
        int xx = dx[i] + x, yy = dy[i] + y;
        if (0 <= xx && xx < 5 && 0 <= yy && yy < 5) {
            mat[xx][yy] = mat[xx][yy] == '0' ? '1' : '0';
        }
    }
}
int check() { // 在mat[0]行固定的情况下，需要多少个操作
    int cnt = 0;
    for (int i = 1; i < 5; ++ i) {
        for (int j = 0; j  < 5; ++ j) {
            if (mat[i-1][j] == '0') {
                turn(i, j);
                cnt ++;
            }
        }
    }
    for (int i = 0; i < 5; ++ i) {
        if (mat[4][i] == '0') return 10;
    }
    return cnt;
}
// 目前操作了idx个，从start开始，需要操作level个灯
void dfs(int idx, int start, int level) { 
    if (idx == level) {
        for(int i = 0; i < level; ++ i) {
            turn(0, arr[i]);
        }
        res = min(res, check() + level);
        memcpy(mat, back, sizeof back);
        return;
    }
    
    for (int i = start; i < 5; ++ i) {
        arr[idx] = i;
        dfs(idx + 1, i + 1, level);
    }
}

int main() {
    int T;
    cin >> T;
    while (T--) {
        for (int i = 0; i < 5; ++ i) {
            scanf("%s", mat[i]);
        }
        memcpy(back, mat, sizeof mat);
        res = 1 << 20;
        for (int i = 0; i < 5; ++ i) {
            dfs(0, 0, i);
        }
        if (res <= 6) {
            cout << res << endl;
        } else {
            cout << -1 << endl;
        }
    }
}