Acwing 843 n-皇后问题

思路：

固定行 找对应列以及对角线和反对角线看能不能符合题意放棋子

code:

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 10;
int n;
char g[N][N]; // 地图
bool row[N], col[N], dg[N], udg[N]; // 分别代表横 竖 正对角线 反对角线

void dfs(int u) {
    if (u == n) {
        for (int i = 0; i < n; i ++ ) {
            for (int j = 0; j < n; j ++ ) {
                cout << g[i][j];
            }
            cout << "\n";
        }
        cout << "\n";
    }
    
    for (int i = 0; i < n; i ++ ) {
        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);
            // 恢复现场
            col[i] = 0;
            dg[u + i] = 0;
            udg[n - u + i] = 0;
            // 地图恢复
            g[u][i] = '.';
        }
    }
}

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;
}