AcWing 843. n-皇后问题
原题链接
中等
作者:
跟着灿哥学切菜
,
2021-01-17 08:51:27
,
所有人可见
,
阅读 239
#include <iostream>
using namespace std;
const int N = 10;
char g[N][N];
bool col[N], dg[N * 2], udg[N * 2];
int n;
void dfs(int u) {
if (u == n) {
//此时使用这种方式来放置每一行的内容
for (int i = 0; i < n; i ++) puts(g[i]);
puts("");
return;
}
for (int i = 0; i < n; i ++) {
if (!col[i] && !dg[u - i + n] && !udg[u + i]) {
g[u][i] = 'Q';
col[i] = dg[u - i + n] = udg[u + i] = true;
dfs(u + 1);
col[i] = dg[u - i + n] = udg[u + i] = false;
g[u][i] = '.';
}
}
}
int main() {
cin >> n;
for (int i = 0; i < n; i ++) {
for (int j = 0; j < n; j ++) {
g[i][j] = '.';
}
}
dfs(0);
return 0;
}
//下面是第二种写法:
#include <iostream>
using namespace std;
const int N = 10;
char g[N][N];
bool row[N], col[N], dg[N * 2], udg[N * 2];
int n;
void dfs(int x, int y, int s) {
//如果此时走到某一行的右边界的时候, 此时需要掉头
if (y == n) {
y = 0;
x ++;
}
if (x == n) {
//说明此时已经到了最后一行的下面一行,表示已经搜索结束
if (s == n) {
for (int i = 0; i < n; i ++) puts(g[i]);
puts("");
}
return;
}
//如果此时没有进行选择
dfs(x, y + 1, s);
//如果此位置放置皇后,但是需要来满足一定的条件才能来放皇后
if (!row[x] && !col[y] && !dg[x + y] && !udg[x - y + n]) {
g[x][y] = 'Q';
row[x] = col[y] = dg[x + y] = udg[x - y + n] = true;
//此时为什么不能使用s ++????
dfs(x, y + 1, s + 1);
row[x] = col[y] = dg[x + y] = udg[x - y + n] = false;
g[x][y] = '.';
}
}
int main() {
cin >> n;
for (int i = 0; i < n; i ++) {
for (int j = 0; j < n; j ++) {
g[i][j] = '.';
}
}
dfs(0, 0, 0);
return 0;
}