n皇后问题(DFS)

原题详细如下:
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皇后问题

算法小白的学习笔记。

你可能感兴趣的:(算法,数据结构)