public List<List<String>> solveNQueens(int n) {
char[][] board = new char[n][n];
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
board[i][j] = '.';
}
}
List<List<String>> res = new ArrayList<>();
dfs(board, 0, res);
return res;
}
//从左到右(第一列---最后一列) 安排好 皇后
private void dfs(char[][] board, int colIndex, List<List<String>> res) {
if (colIndex == board.length) {
res.add(construct(board));
return ;
}
for (int i = 0; i < board.length; i++) {
if (validate(board, i, colIndex)) {
board[i][colIndex] = 'Q';
dfs(board, colIndex+1, res);
board[i][colIndex] = '.';
}
}
}
private boolean validate(char[][] board, int x, int y) {
for (int i = 0; i < board.length; i++) {
for (int j = 0; j < y; j++) {
// board[i][j] 的左上45度角上 正左方向 左下45度角上 不应该有皇后 存在
// 即 (y - j) / (x-i) == -1 x == i (y - j) / (x - i) == 1
if (board[i][j] == 'Q' && (x + j == y + i || x + y == i + j || x == i))
return false;
}
}
return true;
}
private List<String> construct(char[][] board) {
List<String> res = new ArrayList<>();
for (int i = 0; i < board.length; i++) {
res.add(String.valueOf(board[i]));
}
return res;
}
51. N-Queens
最新推荐文章于 2021-09-24 09:40:27 发布