LeetCode N皇后 & N皇后 II

题目链接：https://leetcode-cn.com/problems/n-queens/

题目链接：https://leetcode-cn.com/problems/n-queens-ii/

题目大意：

　　略。

分析：

　　https://blog.csdn.net/kai_wei_zhang/article/details/8033194。

代码如下：

class Solution {
public:
    int allOne;
    vector<vector<string>> ans;
    vector<string> ret;
    string tmp;
    int N;
    
    vector<vector<string>> solveNQueens(int n) {
        allOne = (1 << n) - 1;
        ans.clear();
        tmp.assign(n, '.');
        ret.assign(n, tmp);
        N = n;
        
        solve(0, 0, 0, 0);
        
        return ans;
    }
    
    // vd : 竖直方向
    // ld : 左斜线方向
    // rd : 右斜线方向
    void solve(int level, int vd, int ld, int rd) {
        if(level == N) {
            ans.push_back(ret);
            return;
        }
        
        int limit = (vd | ld | rd) ^ allOne;
        int pos;
        
        while(limit) {
            pos = lowbit(limit);
            limit = cutLowbit(limit);
            
            ret[level][N - __builtin_ffs(pos)] = 'Q';
            solve(level + 1, vd | pos, ((ld | pos) >> 1) & allOne, ((rd | pos) << 1) & allOne);
            ret[level][N - __builtin_ffs(pos)] = '.';
        }
    }
    
    int lowbit(int x) {
        return x & (-x);
    }
    
    int cutLowbit(int x) {
        return x & (x - 1);
    }
};

class Solution {
public:
    int allOne;
    int ans;
    int N;
    
    int totalNQueens(int n) {
        allOne = (1 << n) - 1;
        ans = 0;
        N = n;
        
        solve(0, 0, 0, 0);
        
        return ans;
    }
    
    // vd : 竖直方向
    // ld : 左斜线方向
    // rd : 右斜线方向
    void solve(int level, int vd, int ld, int rd) {
        if(level == N) {
            ++ans;
            return;
        }
        
        int limit = (vd | ld | rd) ^ allOne;
        int pos;
        
        while(limit) {
            pos = lowbit(limit);
            limit = cutLowbit(limit);
            
            solve(level + 1, vd | pos, ((ld | pos) >> 1) & allOne, ((rd | pos) << 1) & allOne);
        }
    }
    
    int lowbit(int x) {
        return x & (-x);
    }
    
    int cutLowbit(int x) {
        return x & (x - 1);
    }
};