【LeetCode】52.N-Queens II

用的是回溯法

class Solution {
public:
    int INITIAL = -1000;
    // 一维数组a[n]中，下标代表行号，对应里面存的值代表列号。比如a[2] = 4，说明2行4列（0行0列为最开始的行列号）
    bool place(int *a, int n, int row, int col)    // 判断第row行第col列可否放置皇后
    {
        for (int i = 0; i < n; ++i)         // 逐行搜索有无冲突，i代表i行
        {
            if (a[i] == col || abs(i - row) == abs(a[i] - col))
                return 0;
        }
        return 1;
    }
    
    int totalNQueens(int n) {
        int a[n];
        for (int i =0; i < n; ++i)      // 数组初始化
            a[i] = INITIAL;
        int row = 0, col = 0;
        int sum = 0;        // 有效的N皇后解
        
        while (row < n)
        {
            while (col < n)     // 对第row行的每一列进行探测，看能否放置皇后
            {
                if (place(a, n, row, col))
                {
                    a[row] = col;
                    col = 0;
                    break;      // 此处break表示第row行已经放置好皇后，可以进入到下一行进行搜索
                }
                else
                    ++col;
            }
            if (a[row] == INITIAL)      // 此时表示第row行没有找到可以放置皇后的位置
            {
                if (row == 0)     // 回溯的终止条件
                    break;
                else
                {
                    --row;
                    col = a[row] + 1;       // 把上一行的皇后往后移一列（还需判断后移后是否满足条件）
                    a[row] = INITIAL;
                    continue;
                }
            }
            if (row == n - 1)       // 程序能进入此判断内，则说明最后一行找到了皇后的位置，产生一个有效解
            {
                ++sum;
                col = a[row] + 1;   // 把这个有效解的位置清除掉，从最后一行、后移一列的位置继续探测
                a[row] = INITIAL;
                continue;
            }
            ++row;
        }
        return sum;
    }
};

中间犯了些小问题，这个题目最好在纸上把逻辑写下

9 / 9 test cases passed.	Status:  Accepted
Runtime: 6 ms	Submitted: 0 minutes ago