acwing843 n-皇后问题(dfs)

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

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

输入格式

共一行，包含整数 n。

输出格式

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

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

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

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

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

数据范围

1≤n≤9

输入样例：

4

输出样例：

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

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

 解法一：按行搜索

#include <iostream>
const int N=20;
bool col[N],dg[N],udg[N];//起标记作用
int n;
char g[10][10];
void dfs(int u)
{
    if (u==n) 
    {
        for (int i=0;i<n;i++)
            puts(g[i]);//相当于cout<<g[i]<<endl;
        puts(" ");//换行
        return;
    }
    for (int i=0;i<n;i++)
    {
        if (!col[i]&&!dg[u+i]&&!udg[u-i+n])
        {
            g[u][i]='Q';
            col[i]=dg[u+i]=udg[u-i+n]=true;//标记
            dfs(u+1);
            col[i]=dg[u+i]=udg[u-i+n]=false;//恢复现场
            g[u][i]='.';
        }
    }
}
int main()
{
    scanf ("%d",&n);
    for (int i=0;i<n;i++)
    for (int j=0;j<n;j++)
    g[i][j]='.';
    dfs(0);
    return 0;
}

解法二：逐个搜索

#include <iostream>
const int N=20;
bool row[N],col[N],dg[N],udg[N];
int n;
char g[10][10];
void dfs(int x,int y,int s)
{
    if (y==n) y=0,x++;
    if (x==n)
    {
        if (s==n)
        {
            for (int i=0;i<n;i++)
            puts(g[i]);
            puts(" ");
        }
        
        return;
    }
    //放皇后
        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]='.';
        }
    //不放皇后
        dfs(x,y+1,s);
}
int main()
{
    scanf ("%d",&n);
    for (int i=0;i<n;i++)
    for (int j=0;j<n;j++)
    g[i][j]='.';
    dfs(0,0,0);
    return 0;
}