一刷83-回溯算法-51N皇后(h)

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

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

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

在这里插入图片描述

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

输入:n = 1
输出:[["Q"]]
-------------------------------
思考:
都知道n皇后问题是回溯算法解决的经典问题,
但是用回溯解决多了组合、切割、子集、排列问题之后,遇到这种二维矩阵还会有点不知所措。

首先来看一下皇后们的约束条件:

不能同行
不能同列
不能同斜线
确定完约束条件,来看看究竟要怎么去搜索皇后们的位置,其实搜索皇后的位置,可以抽象为一棵树。

下面我用一个 3 * 3 的棋盘,将搜索过程抽象为一棵树,如图:

在这里插入图片描述

从图中,可以看出,二维矩阵中矩阵的高就是这棵树的高度,矩阵的宽就是树形结构中每一个节点的宽度。

那么我们用皇后们的约束条件,来回溯搜索这棵树,只要搜索到了树的叶子节点,说明就找到了皇后们的合理位置了

回溯三部曲:
递归函数参数
我依然是定义全局变量二维数组result来记录最终结果。

参数n是棋盘的大小,然后用row来记录当前遍历到棋盘的第几层了。

代码如下:

vector<vector<string>> result;
void backtracking(int n, int row, vector<string>& chessboard) {

递归终止条件
当递归到棋盘最底层(也就是叶子节点)的时候,就可以收集结果并返回了。
代码如下:

if (row == n) {
    result.push_back(chessboard);
    return;
}
单层搜索的逻辑
递归深度就是row控制棋盘的行,每一层里for循环的col控制棋盘的列,一行一列,确定了放置皇后的位置。

每次都是要从新的一行的起始位置开始搜,所以都是从0开始。

代码如下:

for (int col = 0; col < n; col++) {
    if (isValid(row, col, chessboard, n)) { // 验证合法就可以放
        chessboard[row][col] = 'Q'; // 放置皇后
        backtracking(n, row + 1, chessboard);
        chessboard[row][col] = '.'; // 回溯,撤销皇后
    }
}
验证棋盘是否合法
按照如下标准去重:

不能同行
不能同列
不能同斜线 (45度和135度角)
代码如下:

bool isValid(int row, int col, vector<string>& chessboard, int n) {
    int count = 0;
    // 检查列
    for (int i = 0; i < row; i++) { // 这是一个剪枝
        if (chessboard[i][col] == 'Q') {
            return false;
        }
    }
    // 检查 45度角是否有皇后
    for (int i = row - 1, j = col - 1; i >=0 && j >= 0; i--, j--) {
        if (chessboard[i][j] == 'Q') {
            return false;
        }
    }
    // 检查 135度角是否有皇后
    for(int i = row - 1, j = col + 1; i >= 0 && j < n; i--, j++) {
        if (chessboard[i][j] == 'Q') {
            return false;
        }
    }
    return true;
}
在这份代码中,细心的同学可以发现为什么没有在同行进行检查呢?

因为在单层搜索的过程中,每一层递归,只会选for循环(也就是同一行)里的一个元素,所以不用去重了。
可以看出,除了验证棋盘合法性的代码,省下来部分就是按照回溯法模板来的
本题是我们解决棋盘问题的第一道题目。

如果从来没有接触过N皇后问题的同学看着这样的题会感觉无从下手,可能知道要用回溯法,但也不知道该怎么去搜

这里我明确给出了棋盘的宽度就是for循环的长度,递归的深度就是棋盘的高度,
这样就可以套进回溯法的模板里了。
大家可以在仔细体会体会!
-----------------------
代码:
class Solution{
	List<List<String>> res = new ArrayList<>();
    LinkedList<String> path = new LinkedList<>();
	public List<List<String>> solveNQueens(int n){
		char[][] chessboard = new char[n][n];//创建棋盘
		for(char[] c : chessboard) Arrays.fill(c,'.');//初始化棋盘,将所有位置赋值为 .
		backTracking(n,0,chessboard);
		return res;
	}
	private void backTracking(int n,int row,char[][] chessboard){
		if(row==n){
            path = new LinkedList<>();
            for(char[] c : chessboard) path.add(String.copyValueOf(c));//将字符数组转为字符串
			res.add(new ArrayList<>(path));
			return;
		}
		for(int col = 0;col<n;col++){
			if(isValid(row,col,n,chessboard)){
				chessboard[row][col] = 'Q';
				backTracking(n,row+1,chessboard);
				chessboard[row][col] = '.';//回溯
			}
		}
	}
	private boolean isValid(int row,int col,int n,char[][] chessboard){
		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--){ // 检查45度对角线
            if(chessboard[i][j]=='Q') return false;
        }
        for(int i = row-1,j = col+1;i>=0 && j<=n-1;i--,j++){// 检查135度对角线
            if(chessboard[i][j]=='Q') return false;
        }
        return true;
	}
}//public static String copyValueOf(char[] data): 返回指定数组中表示该字符序列的字符串

力扣链接

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值