dfs : n-皇后问题

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

现在给定整数n，请你输出所有的满足条件的棋子摆法。
输入格式
共一行，包含整数n。
输出格式
每个解决方案占n行，每行输出一个长度为n的字符串，用来表示完整的棋盘状态。
其中”.”表示某一个位置的方格状态为空，”Q”表示某一个位置的方格上摆着皇后。
每个方案输出完成后，输出一个空行。
数据范围
1≤n≤9
输入样例：
4
输出样例：
.Q…
…Q
Q…
…Q.

…Q.
Q…
…Q
.Q…
思路：其实仔细思考一下，便能发现n皇后问题就是一个全排列的问题(只不过是满足一定条件的排列)。
而dfs的全排列问题可以回顾一下上一篇博客
代码实现：

#include<iostream>
#include<algorithm>
using namespace std;
const int N = 10;
int n;
int p[N];
char g[N][N];
//分别表示：同一列，'/'正对角线， '\'反对角线
bool col[N], dg[N], udg[N];
void dfs(int u)
{
    if(u == n){
        for(int i = 0; i < n; i ++)
            puts(g[i]);
        puts("");
        return;
    }
    //枚举每一行只放一个皇后，所有并不用考虑每一行会有冲突
    for(int i = 0; i < n; i ++){
        //如果[u][i]这个点对应的该列，该正反对角线都没有被放皇后
        //其实dg和udg的下标就是对应的y = -x + b和 y = x + b的截距
        if(!col[i] && !dg[u + i] && !udg[n - u + i]){
            //先放上皇后并对该列和正反对角线进行标记
            g[u][i] = 'Q';
            col[i] = 1;
            dg[u + i] = 1;
            udg[n - u + i] = 1;
            dfs(u + 1);
            //恢复现场，便于回溯
            g[u][i] = '.';
            col[i] = 0;
            dg[u + i] = 0;
            udg[n - u + i] = 0;
        }
    }
}
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;
}