1、代码
如下代码可以直接在VS上跑起来。
#include<iostream>
#include<vector>
#include<string>
using namespace std;
class Solution {
public:
vector<vector<string>> res;
vector<vector<string>> solveNQueens(int n) {
vector<string> board(n, string(n, '.'));
backtrack(board, 0);
return res;
}
void backtrack(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(board, row + 1);
//撤销选择
board[row][col] = '.';
}
}
bool isValid(vector<string>& board, int row, int col) {
int n = board.size();
//检查列中是否有皇后互相冲突
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 < 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;
}
};
int main()
{
int n = 5;
vector<vector<string>> res;
Solution solutrion;
res = solutrion.solveNQueens(n);
for (int i = 0; i < res.size(); i++) {
for (int j = 0; j < res[i].size(); j++) {
cout << res[i][j] << endl;
}
cout << endl;
cout << endl;
}
cout << res.size() << endl;
return 0;
}
2、运行结果
当n=5时,总共有10种情况,如下图所示。