代码随想录算法训练营第三十天 | LeeCode 51. N 皇后

题目链接：51. N 皇后 - 力扣（LeetCode）

class Solution
{
private:
    vector<vector<string>> result;
    void backtracking(int n, int row, vector<string> &chessBoard)
    {
        if (row == n)
        {
            result.push_back(chessBoard);
            return;
        }
        for (int col = 0; col < n; col++)
        {
            if (isValue(row, col, chessBoard, n))
            {
                chessBoard[row][col] = 'Q';
                backtracking(n, row + 1, chessBoard);
                chessBoard[row][col] = '.';
            }
        }
    }

    bool isValue(int row, int col, vector<string> &chessBoard, int n)
    {
        // 检查列
        for (int i = 0; i < n; i++)
        {
            if (chessBoard[i][col] == 'Q')
            {
                return false;
            }
        }

        // 检查45度角
        for (int i = row - 1, j = col - 1; i >= 0 && j >= 0; i--, j--)
        {
            if (chessBoard[i][j] == 'Q')
            {
                return false;
            }
        }

        // 检查135度角
        for (int i = row - 1, j = col + 1; i >= 0 && j < n; i--, j++)
        {
            if (chessBoard[i][j] == 'Q')
            {
                return false;
            }
        }
        return true;
    }

public:
    vector<vector<string>> solveNQueens(int n)
    {
        vector<string> chessBoard(n,string(n,'.'));
        backtracking(n, 0, chessBoard);
        return result;
    }
};

今天偷懒了，三道题挑了一道n皇后来做，确实有思路，但是如何转换为代码是一个问题。特别是斜边的检验。最后用了一个遍历来检验两个斜边，一路比较上去看是否相等就可以了。

然后总体上就是一个回溯的模版。

判断这个位置是否是一个合格的位置，然后就可以进行回溯操作。

最后忽略的一个点导致执行出错的是没有初始化这个vector<string> ,导致isvalue里面判断是否是一个合适的位置时访问到未初始化的元素导致出错，所以我抄了一个初始化的方案，放n个strng，每个string由n个.组成。就是vector<string> chessBoard(n , string(n , ’.');