1. 程式人生 > 其它 >遞推 AcWing 95. 費解的開關

遞推 AcWing 95. 費解的開關

技術標籤:基本運算演算法

思路:

  1. 列舉第一行的所有情況
  2. 列舉完第一行後,固定住第一行,此時第二行按哪個按鈕是確定的。假設第一行某盞燈是0,那麼需要按第二行對應的燈泡來改變第一行。
  3. 同理,操作第三行和第四行
  4. 遍歷第五行看看是否都為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; }