LeetCode 52 N-Queens II

题意：

n皇后问题，与 http://blog.csdn.net/houserabbit/article/details/72758088 不同，本题指数出解的个数，不要求方案。

思路：

可以用一下比LeetCode 51更优化的搜索方法，每放置一个皇后，我们记录它可以攻击哪一列、以及通过斜线攻击到的第一行的某2个位置。这样在放置皇后的时候，是否能放置到某个位置就可以O(1)判断了。

代码：

class Solution {
public:
    int totalNQueens(int n) {
        int *column = new int[n];
        int *left = new int[n * 2];
        int *right = new int[n * 2];
        memset(column, 0, sizeof(column));
        memset(left, 0, sizeof(left));
        memset(right, 0, sizeof(right));
        int ans = 0;
        dfs(0, n, column, left, right, &ans);
        return ans;
    }

private:
    void dfs(int row, int total, int *column, int *left, int *right, int *ans) {
        if (row == total) {
            ++(*ans);
            return;
        }
        for (int i = 0; i < total; ++i) {
            int l = i - row + total - 1, r = i + row;
            if (!column[i] && !left[l] && !right[r]) {
                column[i] = left[l] = right[r] = 1;
                dfs(row + 1, total, column, left, right, ans);
                column[i] = left[l] = right[r] = 0;
            }
        }
    }
};