DFS算法解决经典n皇后问题

项目场景：

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;
}