n皇后问题[DFS]

n皇后问题

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

输入格式： 输一个n（1<=n<=9),代表棋盘尺寸。
输出格式：每个解决方案占n行，每行输出一个长度为n的字符串。
其中”.”表示某一个位置的方格状态为空，”Q”表示某一个位置的方格上摆着皇后。
每组输出顺序任意，保证不重不漏。
每个方案输出完成后，输出一个空行。
输入样例：

4

输出样例：

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

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

题解：对于一条斜率为正的（udg）对角线：y=x+b，截距为b=y-x,遍历过程难免出现y<x的情况，所以加上一个棋盘尺寸n,让b=n+y-x，这样b恒正,我们在处理这个恒正的下标时就方便许多。当然，对于dg对角线b=y+x无需多虑。
思路的话是初始化全局为‘.’（空）。初始化x=0，对每一列去顺序遍历，遇到符合条件的位置就放入Q。放入皇后’Q’的同时，标记这一列和正反对角线为true，表示在这一层循环与往下的循环中，true的地方不能再放‘Q’，然后遍历到x=n的时候代表整个棋盘遍历完成，输出即可。

#include <iostream>
using namespace std;
const int N = 20;
int 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 ++ )//第一列，第二列，第三列...... 
        if (!col[i] && !dg[u + i] && !udg[n - u + i])
        {
            g[u][i] = 'Q';
            col[i] = dg[u + i] = udg[n - u + i] = true;
            dfs(u + 1);//枚举下一行
            col[i] = dg[u + i] = udg[n - u + i] = false;
            g[u][i] = '.';
        }
        //根据性质得出每一行必须有一个Q 
        //如果说i循环结束说明这一行没法放Q
		//则代表方案失败，回溯到上一层 
}

int main()
{
    cin >> n;
    for (int i = 0; i < n; i ++ )
        for (int j = 0; j < n; j ++ )
            g[i][j] = '.';

    dfs(0);//枚举第1行

    return 0;
}

类似题解：棋盘问题
棋盘挑战【DFS】