题目详情
n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。
给你一个整数 n ,返回所有不同的 n 皇后问题 的解决方案。
每一种解法包含一个不同的 n 皇后问题 的棋子放置方案,该方案中 'Q' 和 '.' 分别代表了皇后和空位
示例
示例 1:
输入:n = 4
输出:[[".Q..","...Q","Q...","..Q."],["..Q.","Q...","...Q",".Q.."]]
解释:如上图所示,4 皇后问题存在两个不同的解法
示例 2:
输入:n = 1
输出:[["Q"]]
代码
class Solution {
private int[][] map;
private boolean[] zx, yx; // 左斜方和右斜方
private boolean[] col; // 列是否放置棋子
private List<List<String>> lists;
public List<List<String>> solveNQueens(int n) {
lists = new LinkedList<>();
map = new int[n][n];
zx = new boolean[2*n];
yx = new boolean[2*n];
col = new boolean[n];
dfs(0,n);
return lists;
}
public void dfs(int row, int n) {
List<String> list = new LinkedList<>();
if (row == n) {
for (int i = 0; i < n; i ++) {
StringBuffer str = new StringBuffer();
for (int j = 0; j < n; j++) {
char ch = map[i][j] == 1 ? 'Q' : '.';
str.append(ch);
}
list.add(str.toString());
}
lists.add(list);
return;
}
for (int i = 0; i < n; i++) {
if (!col[y] && !(zx[row + i]) && !(yx[row + n - i])) {
map[row][i] = 1;
col[i] = zx[row + i] = yx[row + n - i] = true;
dfs(row + 1, n);
map[row][i] = 0;
col[i] = zx[row + i] = yx[row + n - i] = false;
}
}
}
}