/**
* 请设计一种算法,解决著名的n皇后问题。这里的n皇后问题指在一个n*n的棋盘上放置n个棋子,
* 使得每行每列和每条对角线上都只有一个棋子,求其摆放的方法数。
给定一个int n,请返回方法数,保证n小于等于15
*/
public class Dfs_4n皇后问题 {
static int n;
static int cnt;
static int[] rec;
public static void main(String[] args) {
n = 4;
rec = new int[4];
dfs(0);
System.out.println(cnt);
}
/**
*
* @param row 当前正在处理的行
*/
private static void dfs(int row) {
if (row == n) {
cnt++;
return;
}
//依次尝试在某列上放一个皇后
for (int col = 0; col < n; col++) {
boolean ok = true;
//检验这个皇后是否和之前已经放置的皇后有冲突
for (int i = 0; i < row; i++) {
if (rec[i] == col || i + rec[i] == row + col || rec[i] - i == col - row) {
ok = false;
break;
}
}
/*=======这里可以认为是剪枝=======*/
//这一行的这一列可以放
if (ok) {
rec[row] = col; //标记
dfs(row + 1); //继续找下一行
// rec[row]=0; //恢复原状,这种解法这里是否恢复状态都行,为什么?
}
}
}
}
leetcode–面试题 08.12. 八皇后
class Solution {
List<List<String>> res = new ArrayList<>();
public List<List<String>> solveNQueens(int n) {
int[] rec = new int[n + 1];
dfs(rec, 1);
return res;
}
private void dfs(int[] rec, int row) {
if (row==rec.length){
List<String> strings = new ArrayList<>();
for (int i = 1; i < rec.length; i++) {
StringBuilder stringBuilder = new StringBuilder();
for (int j = 1; j < rec.length-1; j++) {
stringBuilder.append('.');
}
String string = stringBuilder.toString();
String a = string.substring(0,rec[i]-1) + "Q" + string.substring(rec[i]-1);
strings.add(a);
}
res.add(strings);
return;
}
for (int i = 1; i < rec.length; i++) {
if (check(rec, row, i)) {
rec[row] = i;
dfs(rec, row + 1);
// rec[row] = 0;
}
}
}
private boolean check(int[] rec, int row, int i) {
for (int j = 1; j < row; j++) {
if (rec[j] == i) {
return false;
}
if (j + rec[j] == row + i) {
return false;
}
if (j - rec[j] == row - i) {
return false;
}
}
return true;
}
}