https://leetcode-cn.com/problems/n-queens/
N皇后问题是指在N×N格的国际象棋上摆放八个皇后,使其不能互相攻击,即任意两个皇后都不能处于同一行、同一列或同一斜线上,问有多少种摆法。
思路
1.如何判断(row, col)位置是否能放皇后
(1)使用col存储已经摆放皇后的列索引
(2)sum存储 / 方向的对角线,因为 / 方向的对角线的坐标满足 (x1+y1) = (x2+y2),例如 (0+3)=(1+2)=(2+1)=(3+0)
(3)diff存储 \ 方向的对角线,因为 \ 方向的对角线的坐标满足 (x1-y1) = (x2-y2),例如 (0-0)=(1-1)=(2-2)=(3-3)
2. dfs :遍历第row行的每一列元素。
(1)如果发现能放皇后,则col、sum、diff都加上当前元素( i )的相关值,然后构造出该行的棋盘。继续遍历下一行(row+1),摆放皇后棋。递归结束后,当前row行的放置回退一步(col、sum、diff,result 都移除刚刚加入的元素),继续遍历下一列的位置( i++ )。
(3)直到row==n,则找到一种棋子放置方案。
(4)如果第row行中所有棋子都不能放,当前的dfs达到i=n时,当前遍历结束,返回到上一层继续判断下一列的位置。
class Solution {
private List<List<String>> ans = new ArrayList<>();
private Set<Integer> col = new HashSet<>();
private Set<Integer> sum = new HashSet<>();
private Set<Integer> diff = new HashSet<>();
private int n = 0;
public List<List<String>> solveNQueens(int n) {
this.n = n;
dfs(0,new ArrayList<>());
return ans;
}
void dfs(int row, List<String> result){
if(row == n){
ans.add(new ArrayList<>(result));
return;
}
for(int i = 0;i<n;i++){
if(!col.contains(i) && !sum.contains(row+i) && !diff.contains(row-i)){
col.add(i);
sum.add(row+i);
diff.add(row-i);
String tmp = "";
for(int j = 0;j<n;j++){
tmp = tmp + (i==j?"Q":".");
}
result.add(tmp);
dfs(row+1,result);
col.remove((Integer)i);
sum.remove((Integer)(row+i));
diff.remove((Integer)(row-i));
result.remove(result.size() - 1);
}
}
}
}