LeetCode---N-Queens II

题目大意：给出N个皇后，放在N*N的棋盘上使其彼此间不能相互攻击。找出所有不同的方案数。

算法思想：

1.整体思想如LeetCode---N-Queens

2.只需记录方案数即可，每找到一种方案计数加1.

代码如下：

class Solution {
public:
    int totalNQueens(int n) {
        this->columns=vector<int>(n,0);
        this->main_diag=vector<int>(n*2,0);
        this->anti_diag=vector<int>(n*2,0);
        int C=0;//记录方案数
        dfs(C,0);
        return C;
    }
    void dfs(int &C,int row){
        const int N=columns.size();
        if(row==N){
            ++C;
            return ;
        }
        for(int j=0;j<N;++j){
           bool ok=columns[j]==0&&main_diag[row-j+N]==0&&anti_diag[row+j]==0;
           if(!ok) continue;
           columns[j]=main_diag[row-j+N]=anti_diag[row+j]=1;
           dfs(C,row+1);
           columns[j]=main_diag[row-j+N]=anti_diag[row+j]=0;
        }
    }
private:
  vector<int> columns;
  vector<int> main_diag;
  vector<int> anti_diag;
};