相关推荐recommended
n皇后问题(DFS)
作者:mmseoamin日期:2024-04-30

原题详细如下:

n− 皇后问题是指将 n 个皇后放在 n×n 的国际象棋棋盘上,使得皇后不能相互攻击到,即任意两个皇后都不能处于同一行、同一列或同一斜线上。

n皇后问题(DFS),第1张

现在给定整数 n,请你输出所有的满足条件的棋子摆法。

输入格式

共一行,包含整数 n。

输出格式

每个解决方案占 n 行,每行输出一个长度为 n 的字符串,用来表示完整的棋盘状态。

其中 . 表示某一个位置的方格状态为空,Q 表示某一个位置的方格上摆着皇后。

每个方案输出完成后,输出一个空行。

注意:行末不能有多余空格。

输出方案的顺序任意,只要不重复且没有遗漏即可。

数据范围

1≤n≤9

输入样例:

4

输出样例:

.Q..

...Q

Q...

..Q.

..Q.

Q...

...Q

.Q..

解题思路:

从每一行开始,遍历一这行的所有元素,如果这一行的列,正对角,反对角都没有皇后,就在这个位置放入一个皇后,然后继续向下一行进行搜索。

对角坐标如下图:

n皇后问题(DFS),第2张

(蓝色为正对角,绿色为反对角,

故正对角线的坐标为:当前行+当前列,

反对角坐标为:n-当前行+当前列。

代码如下:

#include
using 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皇后问题

算法小白的学习笔记。