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

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

题目描述

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

回溯法

解题思路

回溯法采用深度有限的搜索策略遍历问题的解空间树，可采用递归方式实现，也可采用非递归方式实现，由于递归方式理解起来较为简单，故本文各方法均不提供递归方式，只提供非递归方式。

代码实现

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

    // 棋盘 当前放置哪个皇后 解的数量
    int q[n + 1], k = 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;
    };

    // 开始放置皇后
    while (k > 0) {
   
        // 第k个皇后尝试下一个位置
        q[k]++;
        // 寻找第k行的下一个可以放置的位置
        while (q[k] <= n && !check(k))q[k]++;
        // 已超过当前行的上限l，回溯，返回上一行
        if (q[k] > n)--k;
        // 如果放置完所有皇后，则记录结果，否则放置下一行
        else k == n ? res++ : q[++k] = 0;
    }
    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], k = 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;
    };

    // 中点位置 当前行最多能放到第几列
    int m = (n + 1) >> 1, l;
    // n是否为奇数
    bool odd = n & 1;

    // 开始放置皇后
    while (k > 0) {
   
        // 第k个皇后尝试下一个位置
        q[k]++;
        // 第一行放置的皇后不能超过中点
        if (k == 1)l = m;
        // n为奇数且第一行放在中间时，第二行不能超过中间
        else if (k == 2 && odd && q[1] == m)l = m - 1;
        // 其它情况可以放到中点右边
        else l = n;
        // 寻找第k行的下一个可以放置的位置
        while (q[k] <= l && !check(k))q[k]++;
        // 已超过当前行的上限l，回溯，返回上一行
        if (q[k] > l)--k;
        // 如果放置完所有皇后，则记录结果，否则放置下一行
        else k == n ? res++ : q[++k] = 0;
    }
    
    return res <<