非常经典的DFS题目n皇后:
n−皇后问题是指将 n 个皇后放在 n×n 的国际象棋棋盘上,使得皇后不能相互攻击到,即任意两个皇后都不能处于同一行、同一列或同一斜线上
现在给定整数 n,请你输出所有的满足条件的棋子摆法
输入:
4
输出:
.Q..
...Q
Q...
..Q.
..Q.
Q...
...Q
.Q..
public class DFS_n皇后 {
//初始化,dg代表右斜线中存不存在皇后,udg代表左斜线存不存在皇后
//res保存最终数据,col代表当前列中存不存在皇后
//N开两倍是因为斜线数量是N*2-1
public static int N = 20, n;
public static char[][] res = new char[N][N];
public static boolean[] col = new boolean[N], dg = new boolean[N], udg =new boolean[N];
public static void main(String[] args) {
//初始化,让res全部用“.”代替
for (char[] i : res) {Arrays.fill(i, '.');}
Scanner in = new Scanner(System.in);
n = in.nextInt();
dfs(0);
}
public static void dfs(int x) {
//和正常DFS一样,枚举到x的深度和n相等就输出结果
if (x == n) {
for (int i = 0; i<n; i++) {
for (int j = 0; j<n; j++){System.out.print(res[i][j]);}
System.out.println();
}
System.out.println();
return;
}
for (int y = 0; y<n; y++){
//当前行不存在皇后且左/右斜线都不存在皇后时,将当前位置保存,这里判断行其实就已经包含判断列存不存在
//并且将当前列和当前左/右斜线都标记为true,代表存在皇后
//x代表列,y代表行
if (!col[y] && !dg[y+x] && !udg[y-x+n]) {
res[x][y] = 'Q';
col[y] = dg[y+x] = udg[y-x+n] = true;
dfs(x+1);
//记得这里递归出来的时候要还原数据,进去是什么样出来就是什么样
col[y] = dg[y+x] = udg[y-x+n] = false;
res[x][y] = '.';
}
}
}
}
左斜线和右斜线的图例:(蓝色左,绿色右)
左计算方式为:y=x+b, b=y-x, 而y-x有可能出现负数,所以我们就+n来进行补足
右计算方式为:y=x-b, b=y+x
代码量很少,思路需要好好理解,今天更新的好早,打游戏去啦~
声明:算法思路来源为y总,详细请见https://www.acwing.com/
本文仅用作学习记录和交流