前言

本文中博主將一步一步地、以正常人的順序思維完成題目——費解的開關，使用的核心方法是遞推與遞歸。

題目

參考題目：費解的開關

詳細的題目信息相信大家都已經知道了，因此這里為了簡潔只展示輸入輸出格式及數據范圍。

(資料圖片僅供參考)

核心思維

本題利用遞推做的核心思想很簡單，即當這個5x5數組的第一行被處理完過后，想要開啟第一行仍然滅著的燈，則必須點擊該燈的下一行的相同位置。

因此，只要確定好第一行如何選擇，其他行也自然確定了，之需要判斷該種情況是否滿足題目條件即可。

如圖所示：

假設我們第一行只點一次，即被藍色X的地方，點完后會變成這樣：

如果我們想讓第一行的第一個、第四個變亮，那么第二行的第一個、第四個就是必點的。

因此，我們只需要枚舉第一行的所有選法，然后就能遞推出整個四方體的選法，最后判斷成是否成立。

寫出數據輸入格式

首先，先在主函數里寫出題目要求的輸入格式。先輸入一個n，隨后進行n次循環，每次循環都讀入25個數據放在一個二維數組arr里。為了傳參的時候方便，我們把二維數組放在外面，像這樣：

int arr[5][5];int main(){int n = 0;scanf("%d", &n);while (n--){int i = 0;for (i = 0; i < 5; i++){int j = 0;for (j = 0; j < 5; j++){scanf("%1d", &arr[i][j]);}}//...}return 0;}

注意：本題在Acwing上數據輸入時，每個數據之間沒有空格，因此要控制scanf每次讀取數據的寬度。代碼中的“//...”代表接下來從此處開始寫。

枚舉第一行的選擇

這里我們使用遞歸的方法，即在1 ~ 5里面選出1 ~ 5個數，每一種選法都是一種第一行的選擇。創建遞歸函數dfs（int step），step代表當前枚舉的位置，在外面創建數組choose代表遞歸時每個位置的狀態，每次枚舉當前位置選或者不選，五個位置都枚舉結束后就代表形成了一種情況，隨后利用判斷函數jud對這種情況進行判斷。

int main(){//...dfs(0);//dfs的位置}return 0;}

int arr[5][5];int choose[5];void dfs(int step){if (step == 5){jud(choose);return;}//選choose[step] = 1;dfs(step + 1);choose[step] = 0;//不選dfs(step + 1);}

判斷情況是否成立（1）

隨后我們進行判斷函數jud的書寫，為了防止同一組數組不同的情況互相影響，我們創建一個臨時的數組 arr，復制arr的信息到其中，隨后對 arr進行操作。

之后創建i和j，分別用于遍歷行和列。

由于i和j的值不同，點燈還是滅燈的個數也不同（因為有可能在邊界）。因此，我們創建一個函數change，用于改變arr【i】【j】周圍能改變的燈的亮滅情況。

void jud(int* choose){int _arr[5][5];memcpy(_arr, arr, 25 * 4);//對第一行進行操作int i = 0;//用于遍歷行int j = 0;//用于遍歷列for (j = 0; j < 5; j++){if (choose[j] == 1)//相當于arr【0】【j】被選擇了{change(_arr, 0, j);//...}}}

實現亮滅改變函數

羅列情況，改變周圍燈的亮滅情況，如果你不想寫這么多的代碼，也可以把剛開始創建的數組改為7x7大小，就可以不用考慮邊界了。

void change(int _arr[5][5], int i, int j){_arr[i][j] = !_arr[i][j];if (i == 0){_arr[i + 1][j] = !_arr[i + 1][j];if (j == 0){_arr[i][j+1] = !_arr[i][j+1];}else if (j == 4){_arr[i][j-1] = !_arr[i][j-1];}else{_arr[i][j - 1] = !_arr[i][j - 1];_arr[i][j + 1] = !_arr[i][j + 1];}}else if (i == 4){_arr[i - 1][j] = !_arr[i - 1][j];if (j == 0){_arr[i][j + 1] = !_arr[i][j + 1];}else if (j == 4){_arr[i][j - 1] = !_arr[i][j - 1];}else{_arr[i][j - 1] = !_arr[i][j - 1];_arr[i][j + 1] = !_arr[i][j + 1];}}else{_arr[i - 1][j] = !_arr[i - 1][j];_arr[i + 1][j] = !_arr[i + 1][j];if (j == 0){_arr[i][j+1] = !_arr[i][j+1];}else if (j == 4){_arr[i][j - 1] = !_arr[i][j - 1];}else{_arr[i][j - 1] = !_arr[i][j - 1];_arr[i][j + 1] = !_arr[i][j + 1];}}}

判斷情況是否成立（2）

因為對第一行的每一次選擇也算走了一步，所以在每種情況下設置一個變量time，記錄當前走了幾步，一旦time超過6，就立馬return。

注意：第一行只有五個數，因此在第一行的選擇中time不可能超過6，因此不需要在對第一行的選擇中進行判斷。

void jud(int* choose){int _arr[5][5];memcpy(_arr, arr, 25 * 4);int time = 0;//對第一行進行操作int i = 0;//用于遍歷行int j = 0;//用于遍歷列for (j = 0; j < 5; j++){if (choose[j] == 1)//相當于arr【0】【j】被選擇了{time++;change(_arr, 0, j);}}//...//對2，3，4，5行進行操作}

隨后對第2，3，4，5行進行選擇，對第二行的選擇次數，是源于第一行選擇完之后還有幾個滅著的燈。

因此，我們對上一行進行遍歷，如果_arr【i-1】【j】==0，就把time+1，同時點一下_arr【i】【j】。

注意：此時，time已經有可能超過6了，因此需要進行判斷。

void jud(int* choose){int _arr[5][5];memcpy(_arr, arr, 25 * 4);int time = 0;//對第一行進行操作int i = 0;//用于遍歷行int j = 0;//用于遍歷列for (j = 0; j < 5; j++){if (choose[j] == 1)//相當于arr【0】【j】被選擇了{time++;change(_arr, 0, j);}}//...//對2，3，4，5行進行操作for (i = 1; i < 5; i++){for (j = 0; j < 5; j++){if (_arr[i - 1][j] == 0){time++;if (time > 6){return;}change(_arr, i, j);}}}//...}

現在，我們已經對1 ~ 5行全部選擇完畢，但是不確定是否全部都為1，因此需要進行遍歷一次，一旦出現為0的情況，說明這種情況不可取，馬上返回。

//檢測數組中是否全部為1for (i = 0; i < 5; i++){for (j = 0; j < 5; j++){if (_arr[i][j] == 0){return;}}}//...//運行到這里，說明此種方案可行

對輸出數據的處理

題目要求我們輸出所有可行的方案中步數最少的一種所消耗的步數，如果沒有可行方案則返回-1。

因此，我們設置一個全局變量min_time并令其初始化為一個大于6的數，一旦出現一個time小于min_time，就把min_time更新為time。

如果還有更小的time，就能再次更新。

int arr[5][5];int choose[5];int min_time = 10;

//運行到這里，說明此種方案可行min_time = time;return;//...

隨后，我們進行最后的處理。

當dfs（0）結束之后，我們得到了一個min_time，因為它的初始值大于6，所以只要有可行方案存在，該值就一定會被改變，否則，它就依然還是原來的值。

所以，我們設置一個if語句，如果該值為10（初始值），代表沒有可行方案，打印-1后換行。

如果該值不等于10，就打印這個數后換行，代表最小步數為該值。

注意：因為min_time是我們在全局定義的，因此打印完了以后不要忘記再將其重新賦值為10哦。（博主改了很久才想到這一點，TAT）

int main(){int n = 0;scanf("%d", &n);while (n--){int i = 0;for (i = 0; i < 5; i++){int j = 0;for (j = 0; j < 5; j++){scanf("%1d", &arr[i][j]);}}dfs(0);if (min_time == 10){printf("-1\n");}else{printf("%d\n", min_time);min_time = 10;}}return 0;}

感謝您的閱讀與耐心~ 如有錯誤煩請指出~

關鍵詞： 編程算法