解法:回溯算法
class Solution {
// 路径:board 中小于 row 的那些行都已经成功放置了皇后
// 选择列表:第 row 行的所有列都是放置皇后的选择
// 结束条件:row 超过 board 的最后一行
void backtrack(vector<vector<string>>& res,vector<string>& board, int row) {
if (row == board.size()) { //结束条件
res.push_back(board);
return;
}
int n = board[row].size();
for (int col = 0; col < n; col++) {
if (!isValid(board, row, col)) // 排除不合法选择
continue;
board[row][col] = 'Q'; // 做选择
backtrack(res,board, row + 1); // 进入下一行决策
board[row][col] = '.'; // 撤销选择
}
}
/* 是否可以在 board[row][col] 放置皇后? */
bool isValid(vector<string>& board, int row, int col) {
int n = board.size();
// 检查列是否有皇后互相冲突
for (int i = 0; i < n; i++) {
if (board[i][col] == 'Q')
return false;
}
// 检查右上方是否有皇后互相冲突
for (int i = row - 1, j = col + 1;
i >= 0 && j < n; i--, j++) {
if (board[i][j] == 'Q')
return false;
}
// 检查左上方是否有皇后互相冲突
for (int i = row - 1, j = col - 1;
i >= 0 && j >= 0; i--, j--) {
if (board[i][j] == 'Q')
return false;
}
return true;
}
public:
vector<vector<string>> solveNQueens(int n) {
vector<vector<string>> res;
vector<string> board(n, string(n, '.'));
backtrack(res,board, 0);
return res;
}
};