AcWing 费解的开关 二进制枚举

 

 

 解题思路：每一行每一个开关是否需要操作完全由上一行灯的亮灭状态所决定。

所以用二进制枚举第一行开关的操作情况。

第一行灯的亮灭状态确定后，整个5*5所有灯的亮灭状态就确定了。

因为最后一行没有下一行了，所以判断最后一行是否全为亮，若不全为亮就是不合题意。

#include <bits/stdc++.h>
using namespace std;
const int N = 6;
char g[N][N], bk[N][N]; 
int dx[5] = {-1, 0, 1, 0, 0};
int dy[5] = {0, 1, 0, -1, 0};
void turn(int x, int y) {
    for (int i = 0; i < 5; i++) {
        int a = x + dx[i];
        int b = y + dy[i];
        if (a < 0 || a >= 5 || b < 0 || b >= 5) {
            continue;
        }
        g[a][b] ^= 1;
    }
}
int main() {
    int T;
    cin >> T;
    while (T--) {
        for (int i = 0; i < 5; i++) {
            cin >> g[i];
        }
        int res = 10;
        for (int op = 0; op < 32; op++) {
            memcpy(bk, g, sizeof g);
            int step = 0;
            for (int i = 0; i < 5; i++) {
                if (op >> i & 1) {
                    step++;
                    turn(0, i);
                }
            }
            for (int i = 0; i < 4; i++) {
                for (int j = 0; j < 5; j++) {
                    if (g[i][j] == '0') {
                         step++;
                         turn(i + 1, j);
                     } 
                }
            }
            bool flag = false;
            for (int i = 0; i < 5; i++) {
                if (g[4][i] == '0') {
                    flag = true;
                    break;
                }
            }
            if (!flag) {
                res = min(res, step);
            }
            memcpy(g, bk, sizeof bk);
        }
        if (res > 6) {
            res = -1;
        }
        cout << res << endl;
    }
    return 0;
}