LeetCode 51 N皇后

题目

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

皇后不能在同一行，同一列，同对角线

思路一 回溯思想

注意：

• 一行只可能有一个皇后且一列也只可能有一个皇后。这意味着没有必要再棋盘上考虑所有的方格。只需要按列循环即可。

思路步骤：

• 写回溯函数 helper(row = 0).

• 从第一个 row = 0 开始.

• 循环列并且试图在每个 column 中放置皇后.

• 判断方格 (row, column) 能不能放Q

• 在 (row, column) 方格上放置Q。

• If 所有的行被考虑过，row == N
  意味着我们找到了一个解

• Else
  继续考虑接下来的皇后放置 backtrack(row + 1).
  回溯：将在 (row, column) 方格的Q移除.

class Solution {
    public List<List<String>> solveNQueens(int n) {  
        List<List<String>> res = new ArrayList<>();
        char[][] board = new char[n][n];
        init(board);
        helper(res, board, 0);
        return res;      
    }
    //初始化这个board，都为 .
    private void init(char[][] board) {
        for (int i = 0; i < board.length; i++) {
            Arrays.fill(board[i], '.');
        }
    }
    private void helper(List<List<String>> res, char[][] board, int rowIndex) {
        if (rowIndex == board.length) {  //递归终止条件,行数列数都是0开始，达到length，就是越界了
            res.add(generate(board));
            return;
        }
        for (int i = 0; i < board.length; i++) { //当前行的每一列都去判断是否能放Q
            if (isAllow(board, rowIndex, i)) {
                board[rowIndex][i] = 'Q';
                helper(res, board, rowIndex + 1);
                board[rowIndex][i] = '.'; //回溯，删除前面的那个动作，就是把加入的Q变成.,然后在上一次继续去尝试寻找答案
            }
        }
    }
    //判断当前位置是否能放置Q
    private boolean isAllow(char[][] board, int rowIndex, int colIndex) {
        for (int i = 0; i < rowIndex; i++) {
            if (board[i][colIndex] == 'Q') {
                return false;
            }
        }
        for (int i = rowIndex - 1, j = colIndex - 1; i >= 0 && j >= 0; i--, j--) {
            if (board[i][j] == 'Q') {
                return false;
            }
        }
        for (int i = rowIndex - 1, j = colIndex + 1; i >= 0 && j < board.length; i--, j++) {
              if (board[i][j] == 'Q') {
                return false;
            }
        }
        return true;
    }
    //将board这个字符串数组变成链表
    private  List<String> generate(char[][] board) {
        List<String> list = new ArrayList<>();
        for (char[] row : board) {
            StringBuilder sb = new StringBuilder();
            for (char c : row) {
                sb.append(c);
            }
            list.add(sb.toString());
        }
        return list;
    }
}

时间复杂度：O(N!). 放置第 1 个皇后有 N 种可能的方法，放置两个皇后的方法不超过 N (N - 2) ，放置 3 个皇后的方法不超过 N(N - 2)(N - 4) ，以此类推。总体上，时间复杂度为O(N!) .
空间复杂度：O(N^2) ,建立一个char[][] 二维数组