【ACWing】95. 费解的开关

题目地址：

https://www.acwing.com/problem/content/97/

你玩过“拉灯”游戏吗？$25$盏灯排成一个$5\times 5$的方形。每一个灯都有一个开关，游戏者可以改变它的状态。每一步，游戏者可以改变某一个灯的状态。游戏者改变一个灯的状态会产生连锁反应：和这个灯上下左右相邻的灯也要相应地改变其状态。我们用数字$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$

首先拉一次灯相当于对矩阵做了二元域${\mathbb{F}}_2$上的加法，这个加法是满足交换律和结合律的，所以可以先考虑对第一行做怎样的操作。在对第一行做好操作之后，第一行的灯的状态就完全决定了第二行应该怎么按。从而我们可以枚举第一行的所有按灯方式，然后求一个能使得所有灯亮并且次数最少的方案即可。代码如下：

#include <iostream>
#include <cstring>
using namespace std;

const int N = 5;
char g[N][N], bg[N][N];

void flip(char& x) {
    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;
        for (int op = 0; op < 32; op++) {
            int cnt = 0;
            memcpy(g, bg, sizeof g);
            for (int i = 0; i < 5; i++) 
                if (op >> i & 1) {
                    click(0, 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);
        }

        printf("%d\n", res > 6 ? -1 : res);
    }

    return 0;
}

时空复杂度$O(1)$。