Leetcode51题-N皇后

n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。

上图为 8 皇后问题的一种解法。

给定一个整数 n，返回所有不同的 n 皇后问题的解决方案。

每一种解法包含一个明确的 n 皇后问题的棋子放置方案，该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。

示例：

输入：4 输出：[ [".Q…", // 解法 1 “…Q”, “Q…”, “…Q.”],

["…Q.", // 解法 2 “Q…”, “…Q”, “.Q…”] ] 解释: 4
皇后问题存在两个不同的解法。

提示：

皇后彼此不能相互攻击，也就是说：任何两个皇后都不能处于同一条横行、纵行或斜线上。

来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/n-queens
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

利用回溯法，每次放置时检查上对角线是否有Q 就可以放置了

class Solution {
public:
    vector<vector<string>>ans;
    vector<vector<string>> solveNQueens(int n) {
    vector<string>res(n,string(n,'.')); 
    dfs(res,0);
    return ans;
    }
    void dfs(vector<string>&res,int row)
    {
       if(row==res.size())//如果遍历完成就加入。
       {
           ans.push_back(res);
       }
       for(int i=0;i<res[0].size();i++)
       {
          if(isCanSetQ(res,row,i))//判断每次放置时 是否有
          {
              res[row][i]='Q';
              dfs(res,row+1);//进入下一行
              res[row][i]='.';//重置  这行下个位置可以再递归
          }
       }
    }
    bool isCanSetQ(vector<string>&res,int row,int col)//判断是否可以放置
    {
        for(int i=0;i<row;i++)
        {
            if(res[i][col]=='Q')return false;//判断该列
        }
        for(int i=row-1,j=col-1;i>=0&&j>=0;j--,i--)
        {
            if(res[i][j]=='Q')return false;//判断左对角线
        }
        for(int i=row-1,j=col+1;i>=0&&j<res[0].size();i--,j++)
        {
            if(res[i][j]=='Q')return false;//判断右对角线
        }
        return true;
    }
};