leetcode 51. N皇后_c++_hard

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

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

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

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

示例:

输入: 4
输出: [
 [".Q..",  // 解法 1
  "...Q",
  "Q...",
  "..Q."],

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

思路：写代码很有意思，回头再补充。

C++代码如下：

class Solution {
    vector<vector<string>> res;
    vector<bool> col,dia1,dia2; //列，两个对角线是否存在
     //n代表的是处理n皇后问题，摆放第index行皇后的位置 row为已经摆放的皇后
    void putQueen(int n,int index,vector<int>& row)
    {
         if( index == n){
            res.push_back(generate(n,row));
            return;
        }
        for(int i=0; i<n;i ++)
        {
            //尝试将第index皇后摆放在第i列
            if( !col[i] && !dia1[index+i] && !dia2[index-i+n-1])
            {
                row.push_back(i);
                col[i]=true;
                dia1[index+i]=true;
                dia2[index-i+n-1]=true;
                putQueen(n,index+1,row);
                //回溯
                col[i]=false;
                dia1[index+i]=false;
                dia2[index-i+n-1]=false;
                row.pop_back();
            }
        }
        return;
    }
    
   vector<string>  generate(int n,vector<int>& row)
   {
       vector<string> res(n,string(n,'.'));
       for(int i=0;i<row.size();i++)
       {
           res[i][row[i]]='Q';
       }
       return res;
   }
    
public:
    vector<vector<string>> solveNQueens(int n) {
       vector<int> row;
        col=vector<bool>(n,false);
         dia1=vector<bool>(n*2-1,false);
         dia2=vector<bool>(n*2-1,false);
        putQueen(n,0,row);
        return res;
    }
};