51. N-Queens

题目:

The n-queens puzzle is the problem of placing n queens on an n x n chessboard such that no two queens attack each other.

Given an integer n, return all distinct solutions to the n-queens puzzle.

Each solution contains a distinct board configuration of the n-queens' placement, where 'Q' and '.' both indicate a queen and an empty space, respectively.

 

Example 1:

51. N-Queens

Input: n = 4
Output: [[".Q..","...Q","Q...","..Q."],["..Q.","Q...","...Q",".Q.."]]
Explanation: There exist two distinct solutions to the 4-queens puzzle as shown above

Example 2:

Input: n = 1
Output: [["Q"]]

 

Constraints:

  • 1 <= n <= 9

 

 

 

思路:

经典N皇后问题,用DFS回溯进行解决。首先初始化一个board,初始内容全为 ' . ',代表了未使用的格子,如果某个格子上要放皇后,我们把它变成字母 ' Q '。在主函数dfs中,我们首先要清楚逻辑小于row的行都已经成功放置了皇后,我们不用管。那么递归base case就是如果row到达了board边界以外,说明已经放置完成了,则这就是一种答案,要将当前board记录进结果中。如果row未达到边界之外,需要继续放置,则对于当前row的每一列进行check,如果不合理就跳过,如果合理就放置Q,并且继续递归,call完dfs函数后进行回溯即可。最后是一个check函数,因为我们是从第一行放置到最后一行,对于当前行来说,之后的行根本不用检查,因为还没放置。所以对于特定的一格,我们只要检查当前列,即这个[i , j]位置上面的所有行即可。这里分成三个方向,左上,右上和正上。正上只要从 i - 1 开始,保持 j 不变直到 i 为0即可;左上从i - 1,j - 1开始,每次都 i--, j--直到 i 为0或者 j 为0即可;同理,右上 左上从i - 1,j + 1开始,每次都 i--, j++直到 i 为0或者 j 为边界即可。

 

 

 

 

代码:

class Solution {

public:

    vector<vector<string>> solveNQueens(int n) {

        vector<string> board(n,string(n,'.'));

        dfs(board,0);

        return res;

    }

private:

    vector<vector<string>> res;

    void dfs(vector<string> &board, int row)

    {

        if(row==board.size())

        {

            res.push_back(board);

            return;

        }

        int n=board[row].size();

        for(int i=0;i<n;i++)

        {

            if(!check(board,row,i))

                continue;

            board[row][i]='Q';

            dfs(board,row+1);

            board[row][i]='.';

        }

    }

    bool check(vector<string>&board,int row, int col)

    {

        int n=board.size();

        for(int i=0;i<row;i++)

        {

             if(board[i][col]=='Q')

                return false;

        }

        for(int i=row-1, j=col-1;i>=0&&j>=0;i--,j--)

        {

            if(board[i][j]=='Q')

                return false;

        }

        for(int i=row-1, j=col+1;i>=0&&j<n;i--,j++)

        {

            if(board[i][j]=='Q')

                return false;

        }

        return true;

    }

 

};

上一篇:Python解决n皇后


下一篇:八皇后问题