N皇后问题求解
今天在面试中被问到了关于8皇后问题的求解,可能面试官觉得太过于常见又改成了N皇后问题求解,自己太长时间没写算法代码也给忘了个干净,在这里回顾一下N皇后问题的回溯算法
8皇后问题求解
提到n皇后问题就不得不考虑八皇后问题,这是最初问题的原型,我们这里是采用递归来实现的
1 要用到递归我们就不得不考虑到递归的因子,也就是我们以哪一个变量为基准进行递归以及如何递归
2 跳出递归的条件
3 问题的剪枝函数也可以说是限制条件
算法思想:从第一行的第一列开始放置第一个皇后,然后第二行找不冲突的位置进行放置,如果整个第二行都没有合适的位置就回溯到第一行,挪动初始放置的皇后的位置,再进入第二行进行放置,
重复上面的过程直到最终递归的行数x>n,就可以跳出递归函数,行x就是我们的递归因子
另外问题的限制条件就是不能同行同列同对角线,前两者都好说,第三个限制很显然可以转化成斜率来表示。
我在实现算法的时候最开始碰到的问题是如何表示某行某列如果放置了元素该如何表示,开始设想的是a[i][j]=1即为该位置放置了元素,但是实际上我们只需要使用一维数组即可完成,对于a[j]来说,j就是当前的行,a[j]中所存的数值就代表列 a[1]=2 就代表皇后放置在第一行的第二列
限制条件的表示若当前行是x,列是y我们就需要找x行之前是否所有已经放置的皇后都不冲突,即有
while j<x
若 a[j]==y||abs(x-j)==abs(y-a[j])
返回假
否则j++
返回真
参考代码
#include<iostream>
#include<cmath>
using namespace std;
const int N=21;
int a[N];
int n,count=0;
void print()
{
for(int i=1;i<=n;i++)
{
cout<<a[i]<<" ";
}
cout<<endl;
}
int try1(int x,int y)
{
int j=1;
while(j<x)
{
if(a[j]==y||abs(x-j)==abs(a[j]-y))
return 0;
j++;
}
return 1;
}
void place(int x)
{
int y;
if(x>n)
{
print();
count++;
}
else
{
for(int y=1;y<=n;y++)
{
if(try1(x,y))
{
a[x]=y;
place(x+1);
}
}
}
}
int main()
{
cin>>n;
place(1);
cout<<"共有"<<count<<"种方法" <<endl;
return 0;
}
测试结果
图片: )