LeetCode 51 N皇后 HERODING的LeetCode之路

tech2022-09-14  106

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

上图为 8 皇后问题的一种解法。

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

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

示例:

输入:4 输出:[ [".Q…", // 解法 1 “…Q”, “Q…”, “…Q.”],

["…Q.", // 解法 2 “Q…”, “…Q”, “.Q…”] ] 解释: 4 皇后问题存在两个不同的解法。

提示:

皇后彼此不能相互攻击,也就是说:任何两个皇后都不能处于同一条横行、纵行或斜线上。

解题思路: 这道题目和我之前解过的2n皇后问题其实思路是一样的,而且之前那个2n皇后问题更加复杂,要考虑两种皇后的问题,这道题目就简单多了,直接一手dfs,判断行列两个对角线是否有重复就完事,没有就放皇后,转到下一行,代码如下:

class Solution { public: static const int M = 20; int row[M], col[M], dg[M], udg[M]; vector<string> a; vector<vector<string>> res;//返回的结果 vector<vector<string>> solveNQueens(int n) { a.resize(n, string(n, '.'));//把a设置成n*n的. dfs(0, n); return res; } void dfs(int u, int n) { if (u == n) { res.push_back(a);//一种方案完成 } for (int i = 0; i < n; i ++) { if (!row[u] && !col[i] && !dg[u - i + n - 1] && !udg[i + u]) {//如果行列对角线都无 a[u][i] = 'Q'; row[u] = col[i] = dg[u - i + n - 1] = udg[i + u] = 1;//标记已经占用过 dfs(u + 1, n); a[u][i] = '.';//一个方案结束后要复原 row[u] = col[i] = dg[u - i + n - 1] = udg[i + u] = 0; } } } };
最新回复(0)