Leetcode No.51 N皇后 java

---

题目

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

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

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

示例 1：

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

示例 2：
输入：n = 1
输出：[[“Q”]]

来源：力扣（LeetCode）
链接：Leetcode No.51 N皇后

---

分析

最暴力的写法就是先把所有摆法全部列出来然后一个一个挑出来，但显然这种方法把很多没必要的情况也算了一遍，肯定有更优化的办法。
首先，n行n列的棋盘有n个皇后，每个皇后横竖和斜方向不可重复其他皇后，则易得出，每一行每一列有且仅有并且必有一个皇后，因此我们可以仅仅用一个n位的int类型的一维数组存储存储皇后位置信息：下标代表行数（0 <= row < n），存储的内容代表列数（0 <= col < n）
其次，我们来看如何判断是否有“皇后冲突”（即是否满足摆法）

1. 因为我们是按行摆皇后，即摆完一行再到下一行去摆，因此行不会存在冲突
2. 因为列满足在 0 到 n 中且没有重复，因此我们取一个列后判断该列是否已经被使用，判断方法：使用集合存储已经使用的列，然后判断现在取的列是否在该集合中。
3. 是否在斜线上的判断方法（重点！！以官方答案为例）：从左上到右下方向，同一条斜线上的每个位置满足行下标与列下标之差相等，例如 (0,0)(0,0) 和 (3,3)(3,3) 在同一条方向一的斜线上。因此使用行下标与列下标之差即可明确表示每一条方向一的斜线。 从右上到左下方向，同一条斜线上的每个位置满足行下标与列下标之和相等，例如 (3,0)(3,0) 和 (1,2)(1,2) 在同一条方向二的斜线上。因此使用行下标与列下标之和即可明确表示每一条方向二的斜线
   这是解题的非常之关键

代码：

class Solution {
    public List<List<String>> solveNQueens(int n) {
        List<List<String>> solve = new ArrayList<>();
        int[] status = new int[n];//皇后位置数组
        Arrays.fill(status, -1);//将数组全部填充为-1，表示还没有皇后
        Set<Integer> columns = new HashSet<>();//初始化列判断的集合
        Set<Integer> diagonals1 = new HashSet<>();//初始化斜方向（左上到右下）判断的集合
        Set<Integer> diagonals2 = new HashSet<>();//初始化斜方向（右上到左下）判断的集合
        backtrack(solve, status, n, 0, columns, diagonals1, diagonals2);//回溯的函数
        return solve;
    }
    
    public void backtrack(List<List<String>> solve, int[] status, int n, int row,
                          Set<Integer> columns, Set<Integer> diagonals1, Set<Integer> diagonals2) {
		//全部摆满后记录结果
        if (row == n) {
            List<String> board = getBoard(status, n);
            solve.add(board);
        } else {
        	//for循环找可以摆的位置
            for (int i = 0; i < n; i++) {
            	//三个if判断是否存在“皇后冲突”，有则直接跳到下一个循环
                if (columns.contains(i))
                    continue;
                int diagonal1 = row - i;
                if (diagonals1.contains(diagonal1))
                    continue;
                int diagonal2 = row + i;
                if (diagonals2.contains(diagonal2))
                    continue;
                //通过则表示可以摆放，摆放后更新相关信息
                columns.add(i);//添加摆放皇后的列
                status[row] = i;//添加摆放的皇后的位置
                diagonals1.add(diagonal1);//添加摆放皇后的斜位置（从左上到右下）
                diagonals2.add(diagonal2);//添加摆放皇后的斜位置（从右上到左下）
                backtrack(solve, status, n, row + 1, columns, diagonals1, diagonals2);//递归进行下一行（row + 1）的摆放
                status[row] = -1;//出了递归后释放该皇后的位置
                columns.remove(i);//释放该皇后占用的列的位置
                diagonals1.remove(diagonal1);//释放该皇后占用的斜位置（从左上到右下）
                diagonals2.remove(diagonal2);//释放该皇后占用的斜位置（从右上到左下）
            }
        }
    }
	//根据皇后位置信息数组（int[] status）来打印摆放的棋盘
    private List<String> getBoard(int[] status, int n) {
        List<String> board = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            char[] temp = new char[n];
            Arrays.fill(temp, '.');
            temp[status[i]] = 'Q';
            board.add(new String(temp));
        }
        return board;
    }
}

---