八皇后问题
标签: DFS
[TOC]
描述
在国际象棋棋盘上放置八个皇后,要求每两个皇后之间不能直接吃掉对方。
正文
DFS是最好写的搜索,这是DFS的经典入门题。值得一提的有两点。
1. 以排或列为单位枚举,每单位只有可能下一枚棋子。
2. 判断某位置是否可以落子的规则是,枚举之前的每一行或列中的棋子,如果棋子与该位置的横坐标之差和纵坐标之差这二者的绝对值之差为0,那么该位置在那个棋子的对角线上,即不能落子。
不假思索地给出代码
#include<iostream>
#include<cmath>
using namespace std;
const int N = 9;
int book[N][N],tot;
bool check(int x,int y)
{
for(int i=1;i<=x;i++){
if(book[i][y]) return false;
for(int j=1;j<=8;j++){
if(book[i][j]){
if(fabs(i-x)-fabs(j-y)==0) return false;
else break;
}
}
}
return true;
}
void dfs(int step)
{
if(step==N){
tot++;
cout<<"No. "<<tot<<endl;
for(int i=1;i<=8;i++){
for(int j=1;j<8;j++)
cout<<book[j][i]<<' ';
cout<<book[8][i]<<endl;
}
return ;
}
for(int i=1;i<=8;i++){
if(check(step,i)){
book[step][i]=1;
dfs(step+1);
book[step][i]=0;
}
}
return ;
}
int main()
{
dfs(1);
return 0;
}