【DFS】n皇后问题

Java基础

浏览数:201

2019-10-8

回溯:  

  递归调用代表开启一个分支,如果希望这个分支返回后某些数据恢复到分支开启前的状态以便重新开始,就要使用到回溯技巧,全排列的交换法,数独,部分和,用到了回溯。下一个状态在开始之前需要利用到之前的状态,此时需要进行回溯,因为之前的状态对现在的状态存在着影响。

剪枝:

  深度优先搜索的时候如果已经明确从当前状态无论如何转移都不会存在(更优)解,就应该终止往下的继续搜索,这种方法叫做剪枝。数独和部分和里面有剪枝:属于隐性的剪枝,当尝试去填一个数字的时候发现某些数字不行那么会把这个这条支路给剪断。当使用if条件的时候把限定条件加进去然后在if里面使用调用dfs,这也是剪枝的一种。高级的剪枝在调用dfs之前对情况进行预判来进行提前剪枝。

题目:n皇后问题

  请设计一种算法,解决著名的n皇后问题。这里的n皇后问题指在一个n*n的棋盘上放置n个棋子,使得每行每列和每条对角线上都只有一个棋子,求其摆放的方法数。给定一个int n,请返回方法数,保证n小于等于15。

  注意:因为题目中只要求求出摆放的方法数,而不需要求解出其中的过程那么我们就不需要进行回溯,如果需要给出摆放的过程那么就需要进行回溯。此外这个题目不需要进行回溯的另外一个原因是回退到平行状态的时候我们不需要考虑同行的棋子对现在退回来的状态的影响,因为位于for循环中我们进入平行状态的同一行的下一列考虑这个位置上是否可以摆放,所以我们不用进行回溯只需要记录其中的摆放的方法数即可。

代码:

 1 public class N皇后问题 {
 2 
 3     static int n;
 4     static int cnt;
 5 
 6     static int[] rec; // 下标代表行,值代表列
 7 
 8     public static void main(String[] args) {
 9         n = 4;
10         rec = new int[4];
11         dfs(0);
12         System.out.println(cnt);
13     }
14 
15     /**
16      * @param row 当前正在处理的行
17      */
18     private static void dfs(int row) {
19         if (row == n) {
20             cnt++;
21             return;
22         }
23         // 关于检查这个点能不能放的问题 想法就是将将所有不能放的地方的值存为-1,发现不太行。
24         // 依次尝试在某列上放一个皇后
25         for (int col = 0; col < n; col++) {
26             boolean ok = true;
27             // 检验这个皇后是否和之前已经放置的皇后有冲突
28             for (int i = 0; i < row; i++) {
29                 // 正对角线 x-y相同,副对角线x+y相同
30                 if (rec[i] == col || i + rec[i] == row + col || rec[i] - i == col - row) {
31                     ok = false;
32                     break;
33                 }
34             }
35             /* =======这里可以认为是剪枝======= */
36             // 这一行的这一列可以放
37             if (ok) {
38                 rec[row] = col; // 标记
39                 dfs(row + 1); // 继续找下一行
40                 // rec[row]=0; //恢复原状,这种解法这里是否恢复状态都行
41             }
42         }
43     }
44 
45 }

结果:

  

  

 

 

作者:|旧市拾荒|