力扣51题题解

n皇后问题：将n个皇后放置在 n×n 的棋盘上，皇后彼此之间不能相互攻击。

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

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

皇后彼此不能相互攻击，也就是说：任何两个皇后都不同行、不同列，也不在同一对角线上。

这里的“对角线”指的是所有的对角线，不只是平分整个棋盘的那两条对角线。

解题思路：
借用树的思维弄清解题步骤：
假定n=3，三个皇后分别为Q1、Q2、Q3
在第一行，有3种放置方法：Q1在第1列、Q1在第2列，Q1在第3列，三种放置方法都是合法的（没有处于同一条横行、纵行或斜线上的Q）
在第二行，以Q1在第1列举例，Q2在第1列时：Q1、Q2同一列，不合法；Q2在第2列时，Q1在Q2左上方，Q1、Q2处于同一斜线，不合法；Q2在第3列时才合法
以此类推，可得出不存在三皇后的情况

首先，定义一个n行n列，保存元素均为’.‘的board，编写回溯函数将每一行中的’.‘依次用’Q’替换，使得每一行都只有一个’Q’

再编写一个检查函数，判断’Q’放置的位置是否合法

最后将n皇后所有的结果保存在容器中，并返回

实现代码：
class Solution {
private:
vector<vector> res;//用于保存n皇后所有结果
//n为棋盘的规格
//row指明当前递归到第几行了
void backtrack(int n, int row, vector &board){//回溯函数，在board的每一行放置一个’Q’
if(n == row){//递归终止条件，遍历到最后一行，说明有n皇后的解决方案，存储此解决方案
res.push_back(board);
return;
}
//在第row行的第col列放置’Q’
for(int col = 0; col < n; col++){
if(isvalid(n, row, col, board)){//判断在board[row][col]处放置’Q’是否合法
board[row][col] = ‘Q’;
backtrack(n, row + 1,board);//递归
board[row][col] = ‘.’;回溯，将’Q’重置为’.’，以便将’Q’放置在同一行的其它位置
}
}
}
//检查’Q’的位置是否合法
bool isvalid(int n, int row, int col, vector &board){
//检查同一列
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;
}
//在逐层递归中，每一行只会有一个’Q’，无须检查同行
return true;
}

public:
vector<vector> solveNQueens(int n) {
vector board(n, string(n, ‘.’));
backtrack(n, 0, board);
return res;
}
};