51. N皇后
思路:在 n*n 的矩阵中摆放 n 个皇后,并且每个皇后不能在同一行,同一列,同一对角线上,求所有的 n 皇后的解。显而易见的是这一题用DFS比较方便,我们只需要四个数组记录每一行、每一列、每一对角线(对角线有两条)上是否已存在元素即可,那么我们如何记录呢?->
最开始我想到的是用 n 乘 n 的4个矩阵来记录,但是这就有一个问题,每次判断每个皇后是否在同一行,同一列,同一对角线上时就需要遍历这四个 n 乘 n 的矩阵,本身用DFS遍历就会比较耗费时间,那么在判断条件上还需要遍历就更加耗费时间了,有没有更好的方法呢?->
如果我们熟悉矩阵的下标,我们就会发现一个规律:如果我们将45 度对角线标记数组的长度设置为为 2 * n - 1,(r, c) 的位置所在的数组下标为 r + c,同样的,135 度对角线标记数组的长度也是 2 * n - 1,(r, c) 的位置所在的数组下标为 (n - 1 - r )+ c。那么这样以来,行和列、以及对角线数组都可以设置为一维数组,每次判断就不要遍历数组,直接访问下标对应的真假即可!ο(=•ω<=)ρ⌒☆
另外还需要注意的一点是,实际上行和列标记数组我们只需要一个即可,比如我们一行一行的摆放,每行只摆放一个即可。完整代码如下:
boolean[] col = null;//因为是一行一行的摆放,每行只摆放一个即可,所以不需要设置row
boolean[] left = null;
boolean[] right = null;
List<List<String>> res = new ArrayList<>();
public List<List<String>> solveNQueens(int n)