Acwing 843.n-皇后问题

题目：Acwing 843.n-皇后问题

描述：

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

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

输入格式

共一行，包含整数 n。

输出格式

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

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

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

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

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

数据范围

1≤n≤9

输入样例：

4

输出样例：

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

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

————————————————————————————————————————————

算法：DFS（深度优先搜索）

解析：该问题可以抽象成从1~n组成的序列集合中筛选出符合某些条件的子集。row[ ]是对每一行是否有棋做标记，col[ ]是对每一列是否有棋做标记，dg[ ]对每个正对角线是否有棋做标记，ndg[ ]对每个负对角线是否有棋做标记。

代码：

#include <iostream>

using namespace std;

const int N = 10;

int seq[N];

bool row[N], col[N], dg[2 * N - 1], ndg[2 * N - 1];

int n;

void dfs(int l)
{
    if(l == n + 1)
    {
        for (int i = 1; i <= n; i ++)
        {
            for(int j = 1; j <= n; j ++)
            {
                if (seq[i] == j) printf("Q");
                else printf(".");
            }
            puts("");
        }
        puts("");
        
        return ;
    }
    
    for (int i = 1; i <= n; i ++)
    {
        if (!(row[i] || col[i] || dg[(l - i + 2 * n - 1) % (2 * n - 1)] || ndg[l + i]))
        {
            seq[l] = i;
            row[i] = col[i] = dg[(l - i + 2 * n - 1) % (2 * n - 1)] = ndg[l + i] = true;
            
            dfs(l + 1);
            
            row[i] = col[i] = dg[(l - i + 2 * n - 1) % (2 * n - 1)] = ndg[l + i] = false;
        }
    }
}

int main()
{
    scanf("%d", &n);
    
    dfs(1);
    
    return 0;
}