LeetCode 51 N 皇后

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
皇后彼此不能相互攻击，也就是说：任何两个皇后都不能处于同一条横行、纵行或斜线上。

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

1.n*n的棋盘里，n个皇后之间不能相碰，即为：同列，左上右下斜线，右上左下斜线，只能有一个Q；

2.可以用int[] queens 记录每行Q所在的列i，比如 row[0]=1，表示第0行，Q所在的列号为1；

3.终止条件 row == n时，达到终止条件，复原棋盘。

class Solution {
    public List<List<String>> solveNQueens(int n) {
        List<List<String>> res = new ArrayList<>();
        //用queens[]记录每行Q所在的列i，用-1做初始化
        int[] queens = new int[n];
        Arrays.fill(queens, -1);
        //记录同列，同斜线已有的Q
        Set<Integer> columns = new HashSet<>();
        Set<Integer> diagonals1 = new HashSet<>();
        Set<Integer> diagonals2 = new HashSet<>();
        backtrace(res, queens, n, 0, columns, diagonals1, diagonals2);
        return res;
    }

    void backtrace(List<List<String>> res, int[] queens, int n, int row, Set<Integer> columns, Set<Integer> diagonals1, Set<Integer> diagonals2) {
        //满足终止条件
        if (row == n) {
            res.add(generateBorad(queens, n));
            return;
        }
        //遍历列
        for (int i = 0; i < n; i++) {
            //排除不符合条件的
            if (columns.contains(i)) {
                continue;
            }
            int diagonal1 = row - i;
            if (diagonals1.contains(diagonal1)) {
                continue;
            }
            int diagonal2 = row + i;
            if (diagonals2.contains(diagonal2)) {
                continue;
            }
            //做选择
            queens[row] = i;//记录每行Q所在的列号
            columns.add(i);
            diagonals1.add(diagonal1);
            diagonals2.add(diagonal2);
            //回溯
            backtrace(res, queens, n, row + 1, columns, diagonals1, diagonals2);
            //撤销选择
            queens[row] = -1;
            columns.remove(i);
            diagonals1.remove(diagonal1);
            diagonals2.remove(diagonal2);
        }
    }

    List<String> generateBorad(int[] queens, int n) {
        List<String> board = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            char[] row = new char[n];
            Arrays.fill(row, '.');//'.'表示空格
            row[queens[i]] = 'Q';//每行Q所在的列
            board.add(new String(row));
        }
        return board;
    }

    @Test
    public void testNQueens() {
        //[[.Q.., ...Q, Q..., ..Q.], [..Q., Q..., ...Q, .Q..]]
        System.out.println(solveNQueens(4));
    }
}