【LeetCode】52. N-Queens II

N-Queens II

Follow up for N-Queens problem.

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

 

简单的回溯法。

cur[i] == j 代表第i行的皇后位于第j列。

对于每一行，尝试所有列，

如果当前行尝试的位置不与前面所有行的位置冲突，则可以递归到下一行。

 

class Solution {
public:
    int totalNQueens(int n) {
        int count = 0;
        vector<int> cur;
        Helper(count, cur, 0, n);
        return count;
    }
    void Helper(int& count, vector<int> cur, int pos, int n)
    {
        if(pos == n)
            count ++;
        else
        {
            for(int i = 0; i < n; i ++)
            {
                cur.push_back(i);
                if(check(cur))
                    Helper(count, cur, pos+1, n);
                cur.pop_back();
            }
        }
    }
    bool check(vector<int> cur)
    {
        int size = cur.size();
        int loc = cur[size-1];
        for(int i = 0; i < size-1; i ++)
        {
            //never same row
            
            //col
            if(cur[i] == loc)
                return false;
            
            //diag
            if(abs(cur[i]-loc) == abs(i-size+1))
                return false;
        }
        return true;
    }
};