算法典型例题：N皇后问题，五种解法，逐步优化（递归版）

本文将介绍N皇后问题的五种解法，包括朴素回溯法、对称优化、标记优化、可用优化、位运算优化，对于每种解题思路，提供相应的递归版代码实现，最后将对每种解法进行测试，横向对比每种解法的求解时间。

题目描述

在 $N\times N$ 格的国际象棋上摆放 $N$ 个皇后，使其不能互相攻击，即任意两个皇后都不能处于同一行、同一列或同一斜线上，问有多少种摆法？

回溯法

解题思路

回溯法采用深度有限的搜索策略遍历问题的解空间树，可采用递归方式实现，也可采用非递归方式实现。本文提供的解题思路均采用递归法实现。

代码实现

/**
 * N皇后问题：回溯法（所有下标均从1开始）
 * @param n 皇后的数量
 * @return 摆法的数量
 */
int queen(int n) {
   

    // 特判
    if (n == 1)return 1;

    // 棋盘  解的数量
    int q[n + 1], res = 0;
    memset(q, 0, sizeof q);

    // 判断第k个皇后放置位置是否合适
    auto check = [&](int k) {
   
        for (int i = 1; i < k; ++i)
            // 同列、同斜线已存在皇后
            if (q[i] == q[k] || abs(q[i] - q[k]) == abs(i - k))return false;
        return true;
    };

    // 尝试放置第i行
    function<void(int)> dfs = [&](int i) {
   
        if (i > n && ++res)return;
        q[i] = 0;
        while (++q[i] <= n) {
   
            if (!check(i))continue;
            dfs(i + 1);
        }
    };

    // 尝试放置第一行
    dfs(1);

    return res;
}

时间复杂度：$O(n^{n^n})$。

空间复杂度：$O(n)$。

对称优化

解题思路

仔细观察N-皇后的解，发现一种方案可以通过“对称”得到另一种方案。以“左右对称”为例，当 $N=5$，限定第一行皇后在左边一半区域时，方案数为 $6$，如图 $1$ 所示。

（N皇后的可行解存在七种对称关系，此处仅讨论左右对称。）

图1

通过“左右对称”可以获得另一方案，同时发现，后面有两种方案重复，去除重复方案后，剩下的刚好是 $N=5$ 时的全部方案，如图 $2$ 所示。

图2

当 $N$ 为偶数时关于中间那条线对称，当 $N$ 为奇数时关于中间那一列对称。利用左右对称可以使得工作量减少一半，为此，在放置皇后时，增加两条限制

1. 第一行的皇后只放在左边一半区域，也即位置小于等于 $(n+1)/2$；
2. 当 $N$ 为奇数且第一行皇后刚好放在 $(n+1)/2$ 位置（即中间）时，为避免重复，第二行皇后必须放在左边一半区域。

代码实现

/**
 * N皇后问题：对称优化（所有下标均从1开始）
 * @param n 皇后的数量
 * @return 摆法的数量
 */
int queen(int n) {
   

    // 特判
    if (n == 1)return 1;

    // 棋盘 中点位置 解的数量
    int q[n + 1], w = (n + 1) >> 1, res = 0;
    memset(q, 0, sizeof q);

    // n是否为奇数
    bool odd = n & 1;

    // 判断第k个皇后放置位置是否合适
    auto check = [&](int k) {
   
        for (int i = 1; i < k; ++i)
            // 同列、同斜线已存在皇后
            if (q[i] == q[k] || abs(q[i] - q[k]) == abs(i - k))return false;
        return true;
    };


    // 尝试放置第i行，第i行放置的上限为k
    function<void(int, int)> dfs = [&](int i, int k) {
   
        if (i > n && ++res)return;
        // 尝试放置第i行的每个位置
        q[i] = 0;
        while (++q[i] <= k) {
   
            if (!check(i))continue;
            // 第1行放中间时，第2行必须放左边
            if