52. N-Queens II (Array; Back-Track)

Follow up for N-Queens problem.

Now, instead outputting board configurations, return the total number of distinct solutions.

class Solution {
public:
    int totalNQueens(int n) {
        result = ;
        if(n==) return result;

        vector<int> flag(n,-); //每行的Q出现在哪列
        backTracking(n, flag, );
        return result;

    }   

    void backTracking(int n, vector<int>& flag, int depth){ //depth is the line number
        if(depth==n){
            result++;
            return;
        }

        for(int i = ; i < n; i++){ //traverse column
            if(check(n,flag, depth, i)) {
                flag[depth] = i;
                backTracking(n,flag,depth+);
                flag[depth] = -; // back track
            }
        }
    }

    bool check(int n, vector<int>& flag, int i, int j){
        for(int k = ; k < i; k++){
            if(flag[k] < ) continue;//no Q in this column
            if(flag[k]==j) return false;//check columns
            if(abs(i-k)==abs(j-flag[k])) return false; //check cross lines
        }
        return true;
    }
private:
    int result;
};