dfs皇后问题
n− 皇后问题是指将 n 个皇后放在 n×n 的国际象棋棋盘上,使得皇后不能相互攻击到,即任意两个皇后都不能处于同一行、同一列或同一斜线上。
输入样例:
4
输出样例:
.Q…
…Q
Q…
…Q.空行…Q.
Q…
…Q
.Q…
对是否对每列col是否使用过进行判断,对对角线与反对角线进行判断来检验该放置是否可行。
下面分析中的(x,y)(x,y)相当于上面的(u,i)(u,i)
1.反对角线 y=x+by=x+b, 截距 b=y−xb=y−x,因为我们要把 bb 当做数组下标来用,显然 bb 不能是负的,所以我们加上 +n+n (实际上+n+4,+2n都行),来保证是结果是正的,即 y - x + n
2.而对角线 y=−x+by=−x+b, 截距是 b=y+xb=y+x,这里截距一定是正的,所以不需要加偏移量
#include <iostream>
using namespace std;
const int N = 20;
// bool数组用来判断搜索的下一个位置是否可行
// col列,dg对角线,udg反对角线
// g[N][N]用来存路径
int n;
char g[N][N];
bool col[N], dg[N], udg[N];
void dfs(int u) {
// u == n 表示已经搜了n行,故输出这条路径
if (u == n) {
for (int i = 0; i < n; i ++ ) puts(g[i]); // 等价于cout << g[i] << endl;
puts(""); // 换行
return;
}
//对n个位置按行搜索
for (int i = 0; i < n; i ++ )
// 剪枝(对于不满足要求的点,不再继续往下搜索)
// udg[n - u + i],+n是为了保证下标非负
if (!col[i] && !dg[u + i] && !udg[n - u + i]) {
g[u][i] = 'Q';
col[i] = dg[u + i] = udg[n - u + i] = true;
dfs(u + 1);
col[i] = dg[u + i] = udg[n - 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;
}
这道题在很早初入门的时候老师给我们上过,现在看起来比之前好理解很多。