温故知新：dfs模板-843. n-皇后问题

n−n−皇后问题是指将 nn 个皇后放在 n×nn×n 的国际象棋棋盘上，使得皇后不能相互攻击到，即任意两个皇后都不能处于同一行、同一列或同一斜线上。

现在给定整数 nn，请你输出所有的满足条件的棋子摆法。

输入格式

共一行，包含整数 nn。

输出格式

每个解决方案占 nn 行，每行输出一个长度为 nn 的字符串，用来表示完整的棋盘状态。

其中 . 表示某一个位置的方格状态为空，Q 表示某一个位置的方格上摆着皇后。

每个方案输出完成后，输出一个空行。

注意：行末不能有多余空格。

输出方案的顺序任意，只要不重复且没有遗漏即可。

数据范围

1≤n≤91≤n≤9

输入样例：

4

输出样例：

.Q..
...Q
Q...
..Q.
 
..Q.
Q...
...Q
.Q..

思路

深度优先搜索，我们需要排除永远不可能的情况（剪枝），首先是初始化二维数组，把二维数组初始化为'.'

    for(int i=0;i
    {
        for(int j=0;j
        {
            g[i][j]='.';
        }
    }

深度优先搜索分两步走，第一步是判断有没有走到终点，走到终点就输出我们需要的答案

    if(u==n)
    {
        for(int i=0;iputs(g[i]);
        puts("");
        return;
    }

第二步是遍历每一行，利用条件判断，找到可以符合条件的情况（该题是行，对角线，反对角线不能被使用过），然后改变使用状态，修改字符数组的内容，递归调用dfs函数，恢复现场，把状态和字符数组的内容都修改回来

    int x=u;
    for(int y=0;y
    {
        if(!col[y]&&!dg[y+x]&&!udg[y-x+n])
        {
            col[y]=dg[y+x]=udg[y-x+n]=true;
            g[x][y]='Q';
            dfs(x+1);
            col[y]=dg[y+x]=udg[y-x+n]=false;
            g[x][y]='.';
        }
    }

这里把u和i更换成了x和y，感觉更加方便理解

代码

#include
using namespace std;
 
int n;
const int N=20;
char g[N][N];
bool col[N],dg[N],udg[N];
 
void dfs(int u)
{
    if(u==n)
    {
        for(int i=0;iputs(g[i]);
        puts("");
        return;
    }
    
    int x=u;
    for(int y=0;y
    {
        if(!col[y]&&!dg[y+x]&&!udg[y-x+n])
        {
            col[y]=dg[y+x]=udg[y-x+n]=true;
            g[x][y]='Q';
            dfs(x+1);
            col[y]=dg[y+x]=udg[y-x+n]=false;
            g[x][y]='.';
        }
    }
}
 
int main()
{
    scanf("%d",&n);
    for(int i=0;i
    {
        for(int j=0;j
        {
            g[i][j]='.';
        }
    }
    dfs(0);
    return 0;
}