这个就属于深度搜索问题了,之前也做过一个深度搜索问题,感觉相同点就是在搜索函数中往往进行递归。关于n皇后问题课本上有一种解法,要适应这个题只需稍加修改。虽然我形式上明白了,但是对如何具体深搜的有的稀里糊涂的。。提交的时候最后一个数据点也就是13皇后问题超时,为1015ms,就超15ms,,,,百度了一下,找到了许多优化方法,但是还在理解之中。
#include<iostream>
#include<cmath>
using namespace std;
int N;
int sum=0;
int queenans[1000];
void nqueen(int n)
{
int i;
if(n==N)//
{
sum++;
if(sum<4){//控制输出前三个
for(i=0;i<N;++i)
{
cout<<queenans[i]<<' ';
}
cout<<endl;
return;
}
}
for(i=1;i<=N;++i)//逐一尝试第n个皇后的位置
{
int j;
for(j=0;j<n;++j)
{//比较已经摆好的前n个皇后的位置,看是否冲突
if(queenans[j]==i||abs(queenans[j]-i)==abs(n-j)){
break;//冲突,比较下一个位置
}
}
if(j==n)//不冲突
{
queenans[n]=i;
nqueen(n+1);//摆放下一个皇后
}
}
}
int main()
{
cin>>N;
nqueen(0);
cout<<sum;
return 0;
}