题目
那年秋招,遇见这道题,坐着如喽啰。
其实就是dfs+回溯,目的是把所有的情况都遍历一遍。dfs的终止条件是填充到最后一行,因为是一行一行的填充。
然后就是判断在某个位置填充的有效性。分别看这个点的左上方和右上方和上方是否存在Q
class Solution {
public List<List<String>> solveNQueens(int n) {
String[][] mat = new String[n][n];
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
mat[i][j] = ".";
}
}
List<List<String>> res = new ArrayList<>();
dfs(res,mat,0,n);
return res;
}
public void dfs(List<List<String>> res,String[][] mat,int pos,int n){
if(pos==n-1){
for(int i=0;i<n;i++){
if(isValid(mat,pos,i)){
mat[pos][i] = "Q";
List<String> list = new ArrayList<>();
for(int j=0;j<n;j++){
String s = "";
for(int k=0;k<n;k++){
s += mat[j][k];
}
list.add(s);
}
res.add(list);
mat[pos][i] = ".";
}
}
return;
}else{
for(int i=0;i<n;i++){
if(isValid(mat,pos,i)){
mat[pos][i] = "Q";
dfs(res,mat,pos+1,n);
mat[pos][i] = ".";
}
}
}
}
public boolean isValid(String[][] mat,int i,int j){
for(int row=0;row<mat.length;row++){
if(mat[row][j].equals("Q")){
return false;
}
}
int left=j,up=i;
while(left>=0 && up>=0){
if(mat[up][left].equals("Q")){
return false;
}
left--;
up--;
}
int right=j,newup=i;
while(right<mat.length && newup>=0){
if(mat[newup][right].equals("Q")){
return false;
}
right++;
newup--;
}
return true;
}
}
写的很丑陋,但是思路基本很清楚。