遞推 AcWing 95. 費解的開關
阿新 • • 發佈:2021-01-14
思路:
- 列舉第一行的所有情況
- 列舉完第一行後,固定住第一行,此時第二行按哪個按鈕是確定的。假設第一行某盞燈是0,那麼需要按第二行對應的燈泡來改變第一行。
- 同理,操作第三行和第四行
- 遍歷第五行看看是否都為1。如果都為1,則達到目標。
#include <iostream>
#include <cstring>
using namespace std;
char g[10][10];
int dirX[5] = {0, 1, -1, 0, 0};
int dirY[5] = {0, 0, 0, 1, -1};
void turn(int x, int y) {
for (int i = 0; i < 5; i ++) {
int tx = x + dirX[i];
int ty = y + dirY[i];
if (tx >= 0 && tx < 5 && ty >= 0 && ty < 5)
g[tx][ty] ^= 1;
}
}
int work() {
int minRes = 999999;
for (int k = 0; k < 1 << 5; k ++ ) {
int res = 0;
char backup[10][10];
memcpy(backup, g, sizeof g);
bool is_successful = true;
for (int j = 0; j < 5; j ++) {
if (k >> j & 1) {
turn(0, j);
res ++;
}
}
for (int i = 0; i < 4; i ++) {
for (int j = 0; j < 5; j ++) {
if (g[i][j] == '0') {
turn(i + 1, j);
res ++;
}
}
}
for (int j = 0; j < 5; j ++) {
if (g[4][j] == '0') {
is_successful = false;
break;
}
}
if (is_successful && minRes > res) {
minRes = res;
}
memcpy(g, backup, sizeof backup);
}
if (minRes > 6) minRes = -1;
return minRes;
}
int main() {
int n;
cin >> n;
while (n --) {
for (int i = 0; i < 5; i ++) cin >> g[i];
cout << work() << endl;
}
return 0;
}