LeetCode_51_N皇后

题目链接

• https://leetcode.cn/problems/n-queens/

题目描述

按照国际象棋的规则，皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。

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

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

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

示例 1：

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

示例 2：

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

提示：

• 1 <= n <= 9

解题思路

回溯法

放置皇后主要有三个限制

• 不能同行
• 不能同列
• 不能同一个斜线

下面以3 * 3的棋盘为例子，说明一下回溯法的使用

回溯三部曲

• 递归函数参数
  • n：棋盘大小
  • row:记录当前遍历到棋盘的第几层
  • path：保存符合条件的结果
  • ans：保存结果的集合
• 递归终止条件
  • 找到叶子节点
• 单层逻辑
  • 递归深度就是row控制棋盘的行，每一层里for循环的col控制棋盘的列，一行一列，确定了防止皇后的位置
  • 每次都要从新的一行的起始位置开始搜，所以都是从0开始

验证棋盘是否合法

• 不能同行
• 不能同列
• 不能同斜线，包括45度和135度角

AC代码

class Solution {
    public List<List<String>> solveNQueens(int n) {
        List<List<String>> ans = new ArrayList<>();
        char[][] chess = new char[n][n];
        for (char[] c : chess) {
            Arrays.fill(c, '.');
        }
        backTracing(n, 0, chess, ans);
        return ans;
    }

    private void backTracing(int n, int row, char[][] chess, List<List<String>> ans) {
        if (row == n) {
            ans.add(Array2List(chess));
            return;
        }
        for (int col = 0; col < n; col++) {
            if (isValid(row, col, n, chess)) {
                chess[row][col] = 'Q';
                backTracing(n, row + 1, chess, ans);
                chess[row][col] = '.';
            }
        }
    }

    private List<String> Array2List(char[][] chess) {
        List<String> list = new ArrayList<>();
        for (char[] c : chess) {
            list.add(String.copyValueOf(c));
        }
        return list;
    }

    private boolean isValid(int row, int col, int n, char[][] chess) {
        //检查列
        for (int i = 0; i < row; i++) {
            if (chess[i][col] == 'Q') {
                return false;
            }
        }
        //检查45度对角线
        for (int i = row - 1, j = col - 1; i >= 0 && j >= 0; i--, j--) {
            if (chess[i][j] == 'Q') {
                return false;
            }
        }
        //检查135度对角线
        for (int i = row - 1, j = col + 1; i >= 0 && j < n; i--, j++) {
            if (chess[i][j] == 'Q') {
                return false;
            }
        }
        return true;
    }
}