上图为 8 皇后问题的一种解法。
给定一个整数 n,返回 n 皇后不同的解决方案的数量。
示例:
输入: 4
输出: 2
解释: 4 皇后问题存在如下两个不同的解法。
[
[".Q…", // 解法 1
“…Q”,
“Q…”,
“…Q.”],
["…Q.", // 解法 2
“Q…”,
“…Q”,
“.Q…”]
]
思路:理解了51题就行,以后再来改
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 int totalNQueens(int n) {
robot(0,n);
return ans.size();
}
}