项目场景:
DFS算法解决n皇后问题的思路分析和代码分享。
问题描述
n−皇后问题是指将 n 个皇后放在 n×n 的国际象棋棋盘上,使得皇后不能相互攻击到,即任意两个皇后都不能处于同一行、同一列或同一斜线上。
现在给定整数 n,请你输出所有的满足条件的棋子摆法。
输入格式
共一行,包含整数 n。
输出格式
每个解决方案占 n 行,每行输出一个长度为 n 的字符串,用来表示完整的棋盘状态。
其中 .
表示某一个位置的方格状态为空,Q
表示某一个位置的方格上摆着皇后。
每个方案输出完成后,输出一个空行。
注意:行末不能有多余空格。
输出方案的顺序任意,只要不重复且没有遗漏即可。
数据范围
1≤n≤9
输入样例:
4
输出样例:
.Q..
...Q
Q...
..Q.
..Q.
Q...
...Q
.Q..
原因分析:
解决n皇后问题的思路关键是确定搜索顺序。
根据题目要求可知 两个皇后不能放在同一行 同一列 同一条对角线 同一条反对角线上
解决方案:
对于此问题的思考方向暂且分为两种实现方法。
1.朴素算法
思考过程:遍历棋盘中的每一格子按照先遍历完一行再遍历下一行 直到每一个格子都被遍历完毕
对于每一个格子来说 无非就两种选择 放or不放
对于不放的情况 只需要遍历下一个格子即可 也就是列坐标加1
对于放的情况 则需要判断条件。
2.简化算法
思考过程:每次遍历一行 一行中只能放一个皇后 下一次搜索的为下一行 需要判断条件来进行剪枝操作 也就是除掉那些不满足条件的解 不再进行遍历直接回溯
解法1:
遍历顺序为棋盘中的每一个格子元素 因此对于每一个格子来说只有两种情况 放和不放
在放皇后的步骤中 需要判断是否满足 行 列 对角线 反对角线上都只有一个元素
时间复杂度为O(2^n2)
#include<iostream>
using namespace std;
const int N =20;//对角线+反对角线有2N条
int n;
char g[N][N];//用来存放路径
bool row[N],col[N],dg[N],udg[N];//row 记录行是否被使用 col记录列是否被使用 dg和udg记录对角线和反对角线是否被使用
//x为当前格子的行坐标 y当前元素的列坐标 s为当前棋盘中的皇后数量
void dfs(int x,int y,int s){
//y=n则一行中出界 此时行元素加1 列元素置为0
if(y==n) y=0,x++;
//x=n 则表示遍历到最深的层次中
if(x==n){
//s=n表示满足要求的皇后数 输出结果
if(s==n){
for(int i = 0;i<n;i++) cout<<g[i]<<endl;
puts("");
}
return;
}
//不放皇后
dfs(x,y+1,s);
//放皇后
if(!row[x] && !col[y] && !dg[x+y] && !udg[x-y+n]){
g[x][y] = 'Q';
row[x] = col[y] = dg[x+y] = udg[x-y+n] = true;
dfs(x,y+1,s+1);
row[x] = col[y] = dg[x+y] = udg[x-y+n] = false;
g[x][y] = '.';
}
}
int main(){
cin>>n;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
g[i][j] = '.';
dfs(0,0,0);
return 0;
}
解法2:
按行遍历 每一行只能有一个皇后 所以剪枝时的判断条件为列和对角线及其反对角线
时间复杂度为O(n!)
#include<iostream>
using namespace std;
const int N =20;//对角线+反对角线有2N条
int n;
char g[N][N];//用来存放路径
bool col[N],dg[N],udg[N];//col记录列是否被使用 dg和udg记录对角线和反对角线是否被使用
void dfs(int u){
//当DFS深度为n 即遍历到子节点时 输出每一行的结果
if(u==n){
for(int i=0;i<n;i++) cout<<g[i]<<endl;
puts("");
return;
}
for(int i=0;i<n;i++){
//剪枝过程 当不满足行 对角线 反对角线不在一条线时 不再继续往下搜索
//i-u+n增加n偏移量防止下标为负数
//利用截距来表示每条对角线和反对角线
if(!col[i] && !dg[i+u] && !udg[i-u+n]){
g[u][i] = 'Q';
col[i] = dg[i+u] = udg[i-u+n] = true;
dfs(u+1);
//恢复现场
col[i] = dg[i+u] = udg[i-u+n] = false;
g[u][i] = '.';
}
}
}
int main(){
cin>>n;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
g[i][j] = '.';
dfs(0);
return 0;
}