刷题随笔 Day3

题目一

题目描述

有一个 n \times nn×n 的国际象棋棋盘（nn 行 nn 列的方格图），请在棋盘中摆放 nn 个受伤的国际象棋皇后，要求：

1. 任何两个皇后不在同一行。
2. 任何两个皇后不在同一列。
3. 如果两个皇后在同一条 45 度角的斜线上，这两个皇后之间行号的差值至少为 3 。

请问一共有多少种摆放方案。

输入描述

输入的第一行包含一个整数 nn。

其中，1 \leq n \leq 101≤n≤10。

输出描述

输出一个整数，表示答案。

解题代码

思路：

这是经典问题N个皇后的变式问题，运用递归回溯算法。

定义一个全局变量cnt记录方案的个数，是为了在调用函数后，也能输出方案数cnt的值，初始化一个长度为n的列表，索引表示皇后落的行数，值表示列数。使用for循环在每一行遍历每一列，寻找符合条件的可以落皇后的位置，设置一个标志m，符合条件则递归调用Queen函数，进入下一行，否则继续遍历下一列。如果某行的所有列均没有符合条件的落皇后的位置，则退出当前的函数，进入上一行的for循环，将上一行的落皇后位置列数后移，直至可以在每一行都有符合条件的落皇后位置为止，算作一个方案。

关键点在于，递归调用后的函数，运行结束后会退出栈，实际上是结束了调用函数的if语句，继续回到栈顶函数的for循环，所以可以把每一个方案都遍历到。比如说找到了最后一行皇后的位置，使cnt+1，return退出函数，此时只是栈顶的函数运行完出栈，当前栈顶的函数是Queen(line,n-1)，for循环会继续往后遍历一列，再寻找有无合适的位置，以此类推，层层往上，最终得到结果。