【Leetcode之回溯】0051 N皇后

来源：力扣（LeetCode）

链接：https://leetcode-cn.com/problems/n-queens

实现语言：Java

文章记录的是第一次做力扣题时的代码，之后做题的代码放在GitHub上了，由于博客中的文章有点多，挨个修改费时费力，博客中写的有关力扣的文章和GitHub中的代码并不完全对应。力扣题GitHub地址

题目

n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。

给你一个整数 n ，返回所有不同的 n 皇后问题 的解决方案。

每一种解法包含一个不同的 n 皇后问题 的棋子放置方案，该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。

示例

输入：n = 4
输出：[[".Q..","...Q","Q...","..Q."],["..Q.","Q...","...Q",".Q.."]]
解释：如上图所示，4 皇后问题存在两个不同的解法。

输入：n = 1
输出：[["Q"]]

提示：

• 1 <= n <= 9
• 皇后彼此不能相互攻击，也就是说：任何两个皇后都不能处于同一条横行、纵行或斜线上。

方法：回溯

使用回溯穷举所有位置上可能填的数字。按行遍历每个位置，一行放置一个皇后。

• 若当前位置不能防止皇后，则结束本次遍历。

• 若遍历到最后一行，且可以放置皇后时，找到一条可以放置所有皇后的路径，结束本次遍历。

• 因为要找出所有可以放置的方案，所以使用回溯遍历所有位置，每次回溯完要把当前位置置为空。

代码实现

class Solution {
    List<List<String>> res = new LinkedList<>();

    public List<List<String>> solveNQueens(int n) {
        // 初始化矩阵
        char[][] arr = new char[n][n];
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                arr[i][j] = '.';
            }
        }
        // 从第一行开始放置皇后
        for (int j = 0; j < n; j++) {
            dfs(0, j, arr); // 访问该节点
            arr[0][j] = '.';  // 将该节点置空
        }

        return res;
    }

    public void dfs(int i, int j, char[][] arr) {
        // 该位置不能放置皇后，结束遍历
        if (!isValid(i, j, arr)) return;
        // 该位置可以放置皇后
        arr[i][j] = 'Q';

        // 找到一条可行的路径，添加到最终结果中
        if (i == arr.length-1) {
            List<String> tmp = new LinkedList<>();
            for (int k = 0; k < arr.length; k++) {
                String s = "";
                for (int l = 0; l < arr.length; l++) {
                    s += arr[k][l];
                }
                tmp.add(s);
            }
            res.add(tmp);
            return;
        }
        // 访问下一行
        for (int k = 0; k < arr.length; k++) {
            dfs(i+1, k, arr);  // 访问该节点
            arr[i+1][k] = '.';  // 将该节点置空
        }

    }

    // 判断位置[i,j]是否能放置皇后
    public boolean isValid(int i, int j, char[][] arr) {
        // 判断同一行是否有皇后
        for (int k = 0; k < arr.length; k++) {
            if (arr[i][k] == 'Q') return false;
        }
        // 判断同一列是否有皇后
        for (int k = 0; k < arr.length; k++) {
            if (arr[k][j] == 'Q') return false;
        }
        // 判断两条斜线是否有皇后
        for (int k=i-1, m=j-1; k >= 0 && m >= 0; k--, m--) {
            if (arr[k][m] == 'Q') return false;
        }
        for (int k=i+1, m=j+1; k < arr.length && m < arr.length; k++, m++) {
            if (arr[k][m] == 'Q') return false;
        }

        for (int k=i-1, m=j+1; k >= 0 && m < arr.length; k--, m++) {
            if (arr[k][m] == 'Q') return false;
        }
        for (int k=i+1, m=j-1; k < arr.length && m >= 0; k++, m--) {
            if (arr[k][m] == 'Q') return false;
        }
        return true;
    }
}

执行用时：16 ms

内存消耗：39.6 MB