原题详细如下:
n− 皇后问题是指将 n 个皇后放在 n×n 的国际象棋棋盘上,使得皇后不能相互攻击到,即任意两个皇后都不能处于同一行、同一列或同一斜线上。
现在给定整数 n,请你输出所有的满足条件的棋子摆法。
输入格式
共一行,包含整数 n。
输出格式
每个解决方案占 n 行,每行输出一个长度为 n 的字符串,用来表示完整的棋盘状态。
其中 . 表示某一个位置的方格状态为空,Q 表示某一个位置的方格上摆着皇后。
每个方案输出完成后,输出一个空行。
注意:行末不能有多余空格。
输出方案的顺序任意,只要不重复且没有遗漏即可。
数据范围
1≤n≤9
输入样例:
4
输出样例:
.Q..
...Q
Q...
..Q.
..Q.
Q...
...Q
.Q..
解题思路:
从每一行开始,遍历一这行的所有元素,如果这一行的列,正对角,反对角都没有皇后,就在这个位置放入一个皇后,然后继续向下一行进行搜索。
对角坐标如下图:
(蓝色为正对角,绿色为反对角,
故正对角线的坐标为:当前行+当前列,
反对角坐标为:n-当前行+当前列。
代码如下:
#includeusing namespace std; const int N = 20; int n; bool y[N], dg[N], udg[N]; //分别对应列,正对角,反对角 char p[N][N]; void dfs(int u) { if (u == n) { //如果成功搜索到最后一行,就说明已经找到了一个方案,就把这个方案输出 for (int i = 0; i < n; i++)cout << p[i] << endl; cout << endl; return; } for (int i = 0; i < n; i++) { //遍历这一行当中得元素,如果这一列以及两个对角都没有皇后,就在这个坐标放入一个皇后 if (!y[i] && !dg[u + i] && !udg[n - u + i]) { p[u][i] = 'Q'; y[i] = dg[u + i] = udg[n - u + i] = true; dfs(u + 1); //放完后继续向下一行搜索 //搜索完之后回溯要把数据还原 p[u][i] = '.'; y[i] = dg[u + i] = udg[n - u + i] = false; } } } int main() { cin >> n; for (int i = 0; i < n; i++) for (int j = 0; j < n; j++) p[i][j] = '.'; dfs(0); return 0; }
DFS解n皇后问题
算法小白的学习笔记。
上一篇:【数据结构】双向链表详解