原题链接:
题解:
做法1:
它的参数
u
表示当前处理的行数。具体流程如下:
如果
u
达到了 n+1,说明已经成功放置了 n 个皇后,此时将当前的棋盘状态输出,然后返回。遍历当前行的每个格子,尝试在该格子放置皇后。检查该位置是否满足条件,即当前列、主对角线和副对角线都没有其他皇后。如果满足条件,将皇后放置在该格子上,并标记相应的列、主对角线和副对角线为已占用。然后递归调用
dfs
处理下一行。递归完成后,回溯到上一层状态,撤销对当前格子的放置,同时撤销相应的标记。
本图转载自AcWing 843. n-皇后问题--图解+代码注释 - AcWing
做法2:
它的参数有三个,分别是当前处理的行数
x
、当前处理的列数y
以及已经放置的皇后数量s
。具体流程如下:
如果
y
达到了 n+1,说明已经成功处理完当前行的所有列,需要将列数重新置为1,并将行数加1。如果
x
达到了 n+1,说明已经成功放置了 n 个皇后,且已经处理完所有行,此时检查已经放置的皇后数量s
是否为 n。如果是,表示找到了一个合法的解,输出当前的棋盘状态。在一个格子有两种情况:①不放皇后;②放皇后。(为啥方法一只考虑了放皇后而方法二两个都要考虑呢?)
不放皇后:直接dfs(x,y+1,s)即可
放皇后:在当前格子
(x, y)
尝试放置皇后。检查该位置是否满足条件,即当前列、主对角线和副对角线都没有其他皇后。如果满足条件,将皇后放置在该格子上,并标记相应的列、主对角线和副对角线为已占用。然后递归调用dfs
处理下一个格子。递归完成后,回溯到上一层状态,撤销对当前格子的放置,同时撤销相应的标记。
(为啥方法一只考虑了放皇后而方法二两个都要考虑呢?)
因为方法一是在每一行 u 中,只放一个皇后,并且保证每一行都有一个皇后,不需要额外考虑“不放皇后”的情况。在每一行的循环中,对每一个位置 i 进行判断,如果该位置可以放皇后(不受攻击),就放置皇后,然后递归到下一行。
而方法二如果在某个位置放置皇后后导致s(皇后个数)最终不能为n,即使否定这个位置还原为.后,由于y不能实现自增,所以dfs会在没找到结果情况下直接结束,不会进行下一轮迭代。因此,我们就要考虑到不放皇后的情况,这样子不会把“路”堵死,会让dfs(x,y+1,s)进行下一轮迭代,考虑多种情况。这种情况可以参考下图绿色部分
代码:
做法1:
#include<bits/stdc++.h>
using namespace std;
const int N = 20;
char g[N][N];
bool c[N], d1[N], d2[N];
int n;
void dfs(int u) {
if (u == n + 1) {
for (int i = 1;i <= n;i++) {
for (int j = 1;j <= n;j++)
cout << g[i][j] << ((j != n) ? "" : "\n");
}
cout << endl;
return;
}
else {
for (int i = 1;i <= n;i++) {
if (!c[i] && !d1[i+u] && !d2[u-i + n]) {
g[u][i] = 'Q';
c[i] = d1[i + u] = d2[u - i + n] = 1;
dfs(u + 1);
c[i] = d1[i + u] = d2[u - i + n] = 0;
g[u][i] = '.';
}
}
}
}
int main() {
cin >> n;
for (int i = 1;i <= n;i++) {
for (int j = 1;j <= n;j++)
g[i][j] = '.';
}
dfs(1);
}
做法2:
#include<bits/stdc++.h>
using namespace std;
const int N = 20;
char g[N][N];
bool r[N], c[N], d1[N], d2[N];
int n;
void dfs(int x, int y, int s) {
if (y == n + 1) y = 1, x++;
if (x == n + 1) {
if (s == n) {
for (int i = 1;i <= n;i++) {
for (int j = 1;j <= n;j++)
cout << g[i][j] << ((j != n) ? "" : "\n");
}
cout << endl;
}
return;
}
//不放皇后
dfs(x, y + 1, s);
//放皇后
if (!r[x] && !c[y] && !d1[x + y] && !d2[y - x + n]) {
g[x][y] = 'Q';
r[x] = c[y] = d1[x + y] = d2[y - x + n] = 1;
dfs(x, 1, s + 1);
r[x] = c[y] = d1[x + y] = d2[y - x + n] = 0;
g[x][y] = '.';
}
}
int main() {
cin >> n;
for (int i = 1;i <= n;i++) {
for (int j = 1;j <= n;j++)
g[i][j] = '.';
}
dfs(1, 1, 0);
}