推荐一个详解:作者:carlsun-2
链接:https://leetcode-cn.com/problems/n-queens/solution/dai-ma-sui-xiang-lu-51-n-queenshui-su-fa-2k32/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
做的多了,其实看着也不复杂了,反而是可以顺着作者思路走了,也不会去想乱七八糟的了。
截了些作者的解析图:
首先来看一下皇后们的约束条件:
- 不能同行
- 不能同列
- 不能同斜线
个人微改:
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
class solution{
List<List<String>> res = new ArrayList<>();
public List<List<String>> solveNQueens(int n){
char [][] chessboard = new char[n][n];
for(char c[] : chessboard) {
Arrays.fill(c,'.');
}
backTrack(n,0,chessboard);
return res;
}
public void backTrack(int n,int row,char[][] chessboard) {
if(row == n) {
res.add(Array2List(chessboard));
return;
}
for(int col=0;col<n;col++) {
if(isValid(row,col,n,chessboard)) {
chessboard[row][col]='Q';
backTrack(n,row+1,chessboard);
chessboard[row][col]='.';
}
}
}
public List<String> Array2List(char[][] chessboard) {
List<String> list = new ArrayList<>();
for(char[] c:chessboard) {
list.add(String.copyValueOf(c));
}
return list;
}
public boolean isValid(int row,int col,int n,char[][] chessboard) {
//列:
for(int i=0;i<n;i++) {
if(chessboard[i][col]=='Q') {
return false;
}
}
for(int i=row-1,j=col-1;i>=0&&j>=0;i--,j--) {
if(chessboard[i][j]=='Q') {
return false;
}
}
for(int i=row-1,j=col+1;i>=0&&j<=n-1;i--,j++) {
if(chessboard[i][j]=='Q') {
return false;
}
}
return true;
}
}
public class N_Queens {
public static void main(String args[]) {
solution hei = new solution();
System.out.println(hei.solveNQueens(4));
}
}