2021-06-15

dfs皇后问题

参考来源：acwing题解1
acwing题解2

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

输入样例：
4

输出样例：
.Q…
…Q
Q…
…Q.空行

…Q.
Q…
…Q
.Q…

对是否对每列col是否使用过进行判断，对对角线与反对角线进行判断来检验该放置是否可行。

下面分析中的(x,y)(x,y)相当于上面的(u,i)(u,i)
1.反对角线 y=x+by=x+b, 截距 b=y−xb=y−x，因为我们要把 bb 当做数组下标来用，显然 bb 不能是负的，所以我们加上 +n+n （实际上+n+4,+2n都行），来保证是结果是正的，即 y - x + n
2.而对角线 y=−x+by=−x+b, 截距是 b=y+xb=y+x，这里截距一定是正的，所以不需要加偏移量

#include <iostream>
using namespace std;
const int N = 20; 

// bool数组用来判断搜索的下一个位置是否可行
// col列，dg对角线，udg反对角线
// g[N][N]用来存路径

int n;
char g[N][N];
bool col[N], dg[N], udg[N];

void dfs(int u) {
    // u == n 表示已经搜了n行，故输出这条路径
    if (u == n) {
        for (int i = 0; i < n; i ++ ) puts(g[i]);   // 等价于cout << g[i] << endl;
        puts("");  // 换行
        return;
    }

    //对n个位置按行搜索
    for (int i = 0; i < n; i ++ )
        // 剪枝(对于不满足要求的点，不再继续往下搜索)  
        // udg[n - u + i]，+n是为了保证下标非负
        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] = '.';

        }
}

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

这道题在很早初入门的时候老师给我们上过，现在看起来比之前好理解很多。