N皇后问题是一个经典的问题,在一个N*N的棋盘上放置N个皇后,每行一个并使其不能互相攻击(同一行、同一列、同一斜线上的皇后都会自动攻击)。
回溯算法也叫试探法,它是一种系统地搜索问题的解的方法。回溯算法的基本思想是:从一条路往前走,能进则进,不能进则退回来,换一条路再试。
思路:1、把棋盘存储为一个数组a[N],数组中第i个元素的值代表皇后第a[i]列位置,这样便可以把问题的空间规模压缩为一维O(N),在判断是否冲突时也很简单,首先每行只有一个皇后,且在数组中只占据一个元素的位置,行冲突就不存在了,其次是列冲突,判断一下是否有a[i]与当前要放置皇后的列j相等即可。至于斜线冲突,通过观察可以发现所有在斜线上冲突的皇后的位置都有规律即它们所在的行列互减的绝对值相等,即| row – i | = | col – a[i] | 。这样某个位置是否可以放置皇后的问题已经解决。
2、遍历所有行和列,满足条件则放皇后Q,如果一行所有位置都不能放,则回溯。
实现代码:
<span style="font-size:14px;">#include<iostream>
#include<cmath>//递归求解 dfs
#include<cstdlib>
using namespace std;
int a[25]={0};//a[i]=j,i代表行,j代表列
int n;
//检验第xx行的yy列上是否可以摆放皇后
bool check(int xx,int yy)
{
int j=1;
while(j<xx) //j是已经放了皇后的行
{
if(a[j]==yy || abs(a[j]-yy)==abs(j-xx)) //第j行的皇后是否在y列或(j,a[j])与(x,y)是否在斜线上
return false;
j++;
}
return true;
}
void dfs(int x,int nn)
{
if(x>nn)//皇后个数=n时
{
for(int i=1;i<=nn;i++)
{
for(int j=1;j<=nn;j++)
{
if(a[i]==j)
cout<<'Q'<<" ";
else
cout<<'X'<<" ";
}
cout<<endl;
}
cout<<endl;
}
else
{
for(int k=1;k<=nn;k++)//试探第x行的每一个列
{
if(check(x,k))//满足条件
{
a[x]=k;
dfs(x+1,nn); //递归总是在成功完成了上次的任务的时候才做下一个任务
}
}
}
}
int main()
{
while(cin>>n) //n个皇后
{
dfs(1,n);
}
return 0;
} </span>
4皇后 5皇后(未完整图)
注:Q为皇后的位置