给定一个整数 n,返回所有不同的 n 皇后问题的解决方案。
每一种解法包含一个明确的 n 皇后问题的棋子放置方案,该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。
示例:
输入: 4
输出: [
[".Q…", // 解法 1
“…Q”,
“Q…”,
“…Q.”],
["…Q.", // 解法 2
“Q…”,
“…Q”,
“.Q…”]
]
解释: 4 皇后问题存在两个不同的解法。
这个问题是标准的dfs问题,其实是隐式图搜索问题,其实很关键的有几点:
1.怎么表示状态
2.怎么设计递归函数参数
3.对角线怎么处理(用到初中数学知识,正反斜线关系)
因为不能同行同列,因此idx设计为列元素,就是idx加1只能是下一列放皇后,然后开始idx皇后可以放在第0到n-1行,故用一个数组来保存当前行数,然后列因为是idx可以保持不重复,又因为不能同行,因此行也必须用状态标志一下,最后处理正反对角线,行其实是i,列是idx,用简单的数学知识再加平移,就可算出反斜线,因为数组下标不能为负。
这道题难度我还是觉得是怎么把状态表示成可以表示的参数,递归的参数设计直接影响到这个题是否好写,还有一个是状态的还原,这一般涉及到递归加循环的方式,这个题必须要还原,还有之前做过的组合求和的问题,就是加上一个数不符合条件,就必须要把这个数撤掉,也就是回溯的本质。
class Solution {
List<List<String>> ans=new ArrayList<>();
int [] path=new int[100];
boolean [] hang=new boolean[100];
boolean [] zhengxie=new boolean[100];//记录每个点正斜线不能放
boolean [] fanxie=new boolean[100];//记录每个点斜线不能放
public void robot(int idx,int n){//idx代表当前列
if(idx>=n){
//拼凑答案
List<String> res=new ArrayList<>();
for(int i=0;i<n;i++){
String tmp="";
for(int j=0;j<n;j++){
if(path[i]==j){
tmp+="Q";
}else{
tmp+=".";
}
}
res.add(tmp);
}
ans.add(res);
return;
}
for(int i=0;i<n;i++){//for代表的是邻接点的状态枚举,表示行
if(!hang[i]&&!zhengxie[i+idx]&&!fanxie[idx-i+n-1]){
path[idx]=i;//记录行
zhengxie[i+idx]=true;
fanxie[idx-i+n-1]=true;
hang[i]=true;
robot(idx+1,n);
//还原现场
hang[i]=false;
zhengxie[i+idx]=false;
fanxie[idx-i+n-1]=false;
}
}
}
public List<List<String>> solveNQueens(int n) {
robot(0,n);
return ans;
}
}