代码随想录回溯算法——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”]]

思路

N皇后约束条件：

1. 不能同行
2. 不能同列
3. 不能同斜线

仍然可以把二维棋盘抽象为一棵树：（以3皇后为例举例说明）

可以看到，二维矩阵的高就是树的高度，矩阵的宽就是树中每个节点宽度

回溯三部曲：

1. 递归函数参数：二维数组result存放结果，参数n记录棋盘的大小（矩阵的长宽），用row记录遍历到第几行了
2. 递归终止条件：就像上面我画的图，递归到叶子节点的时候就收集到了一个结果了

if(row == n){
	result.add(chessboard);
	return;
}

3. 单层搜索逻辑：递归深度就是用row控制棋盘的行，每一层for循环的col控制棋盘的列，一行一列，从而确定皇后的位置
4. 验证棋盘是否合法：即一下三个条件

1、不能同行（这个其实不用检查，因为在每层递归中，只会选择for循环（同一行）中的一个元素，所以不需要去重操作）
2、不能同列
3、不能同斜线

java代码如下：

class Solution {
	List<List<String>> res = new ArrayList<>();
	
	public List<List<String>> solveNQueens(int n){
		char[][] chessboard = new char[n][n];//初始化存储棋盘的二维字符数组
		for(char[] c : chessboard){
			Arrays.fill(c,'.');//初始全部用.来填满棋盘
		} 
		backTrack(n, 0, chessboard);
		return res;
	}
	
	public void backTrack(int n, int row, char[][] chessboard){
		if(row == n){//递归到叶子节点时，收集结果
			res.add(Array2List(chessboard));//将符合格式的结果加入到res中
			return;
		}
		
		for(int col = 0; col < n; col++){//row控制行，col控制列
			if(isValid(row, col, n, chessboard)){
				chessboard[row][col] = 'Q';
				backTrack(n, row+1, chessboard);
				chessboard[row][col] = '.';//回溯回之前的状态
			}
		}
	}
	
	public List Array2List(char[][] chessboard){
		List<String> list = new ArrayList<>();
	
		for(char[] c : chessboard){
			list.add(String.copyValueOf(c));//copyValueOf()是Java中的String方法，用于创建具有给定字符数组的字符串，即，它接受一个字符数组并返回一个字符串。 换句话说，可以说copyValueOf()方法用于将字符数组的值复制到字符串，比如把二维数组中的字符'.','.','.','Q'四个字符拼接成了一个字符串“...Q”并且存在list中,即["...Q"]
		}
		return list;//返回的列表相当于一个满足格式的结果
	}

	public boolean isValid(int row, int col, int n, char[][] chessboard){
		//检查是否同一列，row控制行，col控制列，自己画图理解即可
		for(int i = 0; i < row; i++){
			if(chessboard[i][col] == 'Q'){
				return false;
			}
		}
		
		//检查右下到左上的斜对角线
		for (int i = row-1, j = col-1; i >= 0 && j >= 0; i--, j--) {
            if (chessboard[i][j] == 'Q') {
                return false;
            }
        }
		//检查左下到右上的斜对角线
		for (int i = row-1, j = col + 1; i >= 0 && j <= n-1; i--, j++) {
            if (chessboard[i][j] == 'Q') {
                return false;
            }
        }
        return true;
	}
}