LeetCode 52. N皇后 II

题目描述

52. N皇后 II

解法：基于位运算的回溯（C++）

基于位运算的回溯算法在这里不多讲，可以参考 LeetCode 51. N 皇后

我们对于回溯树上的每个结点计算其有多少条路径，最终汇总到根结点的总数，就是我们希望求解的可行方案数

class Solution {
public:
    int totalNQueens(int n) {
        return helper(n, 0, 0, 0, 0);
    }

    int helper(int n, int row, int col, int diag1, int diag2){
        // 结束条件
        if (row == n)
            return 1;
        else 
        {
            int cnt = 0;
            // (1 << n) - 1 得到一个长度为 n 的全 1 二进制序列
            // col | diag1 | diag2 得到的是不能放置皇后的位置，取反得到可以放置皇后的位置
            // 然后用可以放置皇后的位置序列和全 1 二进制序列做与，1 即表示可选的位置
            int availablePosition = ((1 << n) - 1) & (~(col | diag1 | diag2));
            // 做选择
            while (availablePosition != 0 )
            {
                // 获取最低位的 1 的位置，即最前面可以选择放置皇后的位置
                int position = availablePosition & (-availablePosition);
                // 将最低位的 1 置为 0，即选取该位置放置皇后
                availablePosition = availablePosition & (availablePosition - 1);
                // 进入下一次操作
                cnt += helper(n, row + 1, col | position, (diag1 | position) >> 1, (diag2 | position) << 1);
                // 回溯
            }
            return cnt;
        }
    }
};