acwing843.n皇后问题

题目

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

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

输入格式

共一行，包含整数 n。

输出格式

每个解决方案占 n行，每行输出一个长度为 n的字符串，用来表示完整的棋盘状态。其中 . 表示某一个位置的方格状态为空，Q 表示某一个位置的方格上摆着皇后。每个方案输出完成后，输出一个空行。
注意：行末不能有多余空格。

数据范围：

1≤n≤9

输入样例：

4

输出样例：

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

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

思路：

dfs处理每一层，对于每一层for（i~n）找到可以填充的位置，然后递归进入下一层。递归到最深处返回（u > n），返回上一层时候要将当前层做的标记、处理 进行恢复

代码：

#include<bits/stdc++.h>
using namespace std;
const int N = 1e3 + 10;
int n;
char s[N][N];
int row[N],col[N],undiag[N],diag[N];

void dfs(int u){
    if(u > n) {
        for(int i =1 ;i <= n;i ++){
            for(int j =1; j <= n;j ++) cout<<s[i][j];
        cout<<"\n";
        }
        cout<<"\n";
        return ;
    }
    for(int i = 1;i <=n;i ++){
        if(!row[u] && !col[i] && !diag[u+i] && !undiag[u-i + n]){
            s[u][i] = 'Q';
            row[u] = col[i] = diag[u+i] = undiag[u-i+n] = 1;
            dfs(u+1);
            row[u] = col[i] = diag[u+i] = undiag[u-i+n] = 0;
            s[u][i] = '.';
        }
    }
}
int main(){
    cin>>n;
    for(int i =1 ;i <= n;i ++){
        for(int j = 1;j <= n;j ++) s[i][j] = '.';
    }
    dfs(1);
    return 0;
}