八皇后问题(dfs深搜)

题目:(洛谷 P1219)
一个如下的 6×6 的跳棋棋盘,有六个棋子被放置在棋盘上,使得每行、每列有且只有一个,每条对角线(包括两条主对角线的所有平行线)上至多有一个棋子。

八皇后问题(dfs深搜)
上面的布局可以用序列 2 4 6 1 3 5来描述,第 i个数字表示在第 i行的相应位置有一个棋子,如下:
行号 1 2 3 4 5 6
列号 2 4 6 1 3 5
这只是棋子放置的一个解。请编一个程序找出所有棋子放置的解。
并把它们以上面的序列方法输出,解按字典顺序排列。
请输出前 3个解。最后一行是解的总个数。
输入格式
一行一个正整数 nnn,表示棋盘是 n×n大小的。
输出格式
前三行为前三个解,每个解的两个数字之间用一个空格隔开。第四行只有一个数字,表示解的总数。
输入输出样例
输入 #1
6
输出 #1
2 4 6 1 3 5 //第一个解:
3 6 2 5 1 4 //第二个解:
4 1 5 2 6 3 //第三个解:
4 //总的解数:
说明/提示
【数据范围】
对于 100%100%100% 的数据,6≤n≤13。
题解:
皇后是国际象棋中的棋子,可以在横竖左斜右斜八个方向上移动,相同的皇后
不能处在其他皇后的这八个方向上,我们以行为深度进行搜索,这样还需记录列、左斜、右斜三个方向是否能被访问到。左斜的坐标规律:横纵坐标值和为定值。右斜坐标规律:横纵坐标之差为定值。)

代码:
#include<bits/stdc++.h>
using namespace std;

int n,ans,a[20];
bool line[20];//标记列元素是否被访问;
bool L[40],R[40];//标记左斜与右斜元素是否被访问;

void print(){
if(ans<=3){
for(int i=1;i<=n;i++){
cout<<a[i]<<" ";
}
cout<<endl;
}
}
void dfs(int x){
if(x>n){
ans++;
print();
return;
}
for(int i=1;i<=n;i++){
if(!line[i] &&!R[x-i+n] &&!L[x+i] ){
line[i]=1; //标记元素是否被搜索
L[x+i]=1;
R[x-i+n]=1;
a[x]=i; //a[]记录搜索路径
dfs(x+1); //递归
line[i]=0; //回溯
L[x+i]=0;
R[x-i+n]=0;
}
}
}
int main(){
cin>>n;
dfs(1);
cout<<ans<<endl;
return 0;
}
运行结果:
八皇后问题(dfs深搜)
|洛谷提交|:
八皇后问题(dfs深搜)