[leetcode] 51. N-Queens

Then-queens puzzle is the problem of placingn queens on an n×n chessboard such that no two queens attack each other.

Given an integern, return all distinct solutions to then-queens puzzle.
There exist two distinct solutions to the 4-queens puzzle:

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

For example,

[
 [".Q..",  // Solution 1
  "...Q",
  "Q...",
  "..Q."],
 ["..Q.",  // Solution 2
  "Q...",
  "...Q",
  ".Q.."]
]

这道题是经典的八皇后问题，很多教科书中都拿它作为递归回溯的入门题目进行讲解，如果觉得这道题太low跳过即可。题目难度为Hard。

递归回溯的关键问题是找出递归结束条件以及回溯时需要恢复的状态。这里还需要注意对具体位置是否安全的判断，如何判断行、列、对角线是否安全。具体代码：

class Solution {
    void NQueens(vector<vector<string>> &rst, vector<string> &curRst, int row, int n) {
        if(row == n) {
            rst.push_back(curRst);
            return;
        }
        
        for(int i=0; i<n; i++) {
            bool flag = true;
            for(int j=0; j<row; j++) {
                if(curRst[j][i] == 'Q') {
                    flag = false;
                    break;
                }
                if((i+j-row >= 0) && (i+j-row <= n-1) && (curRst[j][i+j-row] == 'Q')) {
                    flag = false;
                    break;
                }
                if((i+row-j >= 0) && (i+row-j <= n-1) && (curRst[j][i+row-j] == 'Q')) {
                    flag = false;
                    break;
                }
            }
            
            if(flag) {
                curRst[row++][i] = 'Q';
                NQueens(rst, curRst, row, n);
                curRst[--row][i] = '.';
            }
        }
    }
public:
    vector<vector<string> > solveNQueens(int n) {
        vector<string> curRst(n, string(n, '.'));
        vector<vector<string>> rst;
        
        NQueens(rst, curRst, 0, n);
        
        return rst;
    }
};

其中函数NQueens表示已经安全放置了前row行的皇后之后，如何放入下一行的皇后。如果row等于N表示已经放完，这就是递归结束条件。然后针对待放入皇后这行的N个位置，依次判断放入其中是否安全，这里判断行、列、对角线安全与否的方法不再详述。如果行、列、对角线都安全，表明这个位置可以放入，然后就可以递归进行下一行。回溯时需要恢复的状态是什么呢？回溯时表明刚才放入的位置不安全，或者已经顺利结束之后回溯到这里，无论怎样，我们要收回刚才放入的皇后，这里需要把row减一 表明这行还没有放皇后 ，然后把 刚才放入皇后的位置从'Q'变回'.' 。这样战场就打扫完毕，可以继续后续的循环遍历了。