N皇后问题

题目描述

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/n-queens

解题思路

N皇后为题可以使用回溯法求解，回溯法有三个基本组成部分，路径、选择列表和结束条件。当满足结束条件是将路径加入，并从选择列表中选择构成路径的节点，当选择列表中的元素是合法的（一般为没有被访问过的）则将该选择节点加入到路径中，之后开启下一层的回溯，最后将选择节点从路径中撤销，恢复成原始状态。
代码如下：

List<List<String>> path = new ArrayList<List<String>>();
public List<List<String>> solveNQueens(int n) {
    if (n <= 0) {
        return new ArrayList<List<String>>();
    }
    // 初始化棋盘
    char[][] board = initialBoard(n);
    // 回溯
    backTrack(board, 0);
    return path;

}

// 回溯
private void backTrack(char[][] board, int row) {
    // 结束条件
    if (row == board.length) {
        ArrayList<String> list = new ArrayList<>();
        for (char[] ch : board) {
            list.add(new String(ch));
        }
        // 添加路径
        path.add(list);
        return;
    }

    int cols = board[row].length;

    // 选择列表
    for (int col = 0; col < cols; col++) {
        // 排除不合法选择
        // row之上的所有行都已经排好了皇后，因此只需要和上方进行比较
        if (!isValid(board, row, col)) {
            continue;
        }
        // 进行选择
        board[row][col] = 'Q';
        // 进行下一次决策
        backTrack(board, row + 1);
        // 撤销选择
        board[row][col] = '.';
    }
}

// 判断是否冲突
private boolean isValid(char[][] board, int row, int col) {
    int len = board.length;
    // 检查当前行以上是否有冲突
    for (int i = 0; i < row; i++) {
        if (board[i][col] == 'Q') {
            return false;
        }
    }
    // 检查右上方是否冲突
    for (int i = row - 1, j = col + 1; i >= 0 && j < len; i--, j++) {
        if (board[i][j] == 'Q') {
            return false;
        }
    }
    // 检查左上方是否冲突
    for (int i = row - 1, j = col - 1; i >= 0 && j >= 0; i--, j--) {
        if (board[i][j] == 'Q') {
            return false;
        }
    }
    return true;
}

// 初始化棋盘
private char[][] initialBoard(int n) {
    char[][] board = new char[n][n];
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            board[i][j] = '.';
        }

    }
    return board;
}