前言

本文中博主将一步一步地、以正常人的顺序思维完成题目——费解的开关，使用的核心方法是递推与递归。

题目

参考题目：费解的开关

(相关资料图)

详细的题目信息相信大家都已经知道了，因此这里为了简洁只展示输入输出格式及数据范围。

核心思维

本题利用递推做的核心思想很简单，即当这个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;}

感谢您的阅读与耐心~ 如有错误烦请指出~

标签 编程算法