n皇后问题_递归回溯法

著名n皇后问题

题目为:

n个皇后摆放在N x N的棋盘格中,使得横、竖和两个对角线方向均不会同时出现两个皇后。

解题思路:

利用递归的方法依次查找,回溯再查找

找到第一行的第一个合适位置后,进入第二行查找,找到合适位置放置皇后,进入第三行,如果此时第三行无解,则回到第二行,寻找第二个合适位置,如果第二行没有合适位置,则回到第一行寻找下一个可以放置皇后的位置
这就是回溯

如图4*4回溯
图片来自:n皇后问题_回溯法 也是看了这篇才终于搞明白

4*4棋盘正确位置:

0 1 0 0
0 0 0 1
1 0 0 0
0 0 1 0

0 0 1 0
1 0 0 0
0 0 0 1
0 1 0 0

我的代码:

import java.util.Scanner;

public class Main {
    static int n = 0;//地图大小
    static int map[][] = null;//实际上不需要,只是用来显示地图
    static int[] x = null;// 储存 X 轴坐标

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        //地图大小
        n = 4;
        map = new int[n][n];
        x = new int[n];
        word(0);
    }
    
    /**
     * 递归回溯
     * @param y  棋盘的 Y 轴
     */
    private static void word(int y) {
    //成功找到最后一行  这里应该是用来计数 我拿来打印地图了
        if (y == n) {
            for (int i = 0; i < n; i++) {
                map[x[i]][i] = 1;
            }
            for (int i = 0; i < n; i++) {
                for (int j = 0; j < n; j++) {
                    System.out.print(map[j][i]+" ");
                }
                System.out.println();
            }
            map = new int[n][n];
            System.out.println();
            return;
        }

        for (int i = 0; i < n; i++) {
            x[y] = i;
            //判断纵斜是否有值
            if (TF(y)) {
                //无值,进入下一层
                word(y + 1);
            }
        }
    }

    /**
     * 根据纵坐标 判断横斜是否有值
     * @param y 棋盘的 Y 轴
     * @return 横纵斜是否纯在皇后
     */
    private static boolean TF(int y) {
        for (int i = 0; i < y; i++) {
            //判断列是否有冲突          判断斜边是否有冲突
            if (x[y] == x[i] || Math.abs(x[y] - x[i]) == Math.abs(y - i)) {
                return false;
            }
        }
        return true;
    }
}

寻找解析的过程中也发现了一个可能是最快的方法吧

目前最快的N皇后问题算法!! 虽然是 2006 年的帖子了但是还是牛逼
N皇后的两个最高效的解法 对算法的解析贴

这是评论区一个大佬的解析

我完整地贴一下:
// N Queens Problem
// 试探-回溯算法,递归实现

// sum用来记录皇后放置成功的不同布局数;upperlim用来标记所有列都已经放置好了皇后。
long sum = 0, upperlim = 1;      

// 试探算法从最右边的列开始。
void test(long row, long ld, long rd){
   if (row != upperlim)
   {
  // row,ld,rd进行“或”运算,求得所有可以放置皇后的列,对应位为0,
          // 然后再取反后“与”上全1的数,来求得当前所有可以放置皇后的位置,对应列改为1。
          // 也就是求取当前哪些列可以放置皇后。
  long pos = upperlim & ~(row | ld | rd); 
  while (pos) // 0 -- 皇后没有地方可放,回溯。
  {
 // 拷贝pos最右边为1的bit,其余bit置0。
 // 也就是取得可以放皇后的最右边的列。
 long p = pos & -pos;                                               

 // 将pos最右边为1的bit清零。
                 // 也就是为获取下一次的最右可用列使用做准备,
                 // 程序将来会回溯到这个位置继续试探。
 pos -= p;                            

 // row + p,将当前列置1,表示记录这次皇后放置的列。
 // (ld + p) << 1,标记当前皇后左边相邻的列不允许下一个皇后放置。
 // (ld + p) >> 1,标记当前皇后右边相邻的列不允许下一个皇后放置。
                 // 此处的移位操作实际上是记录对角线上的限制,只是因为问题都化归
                 // 到一行网格上来解决,所以表示为列的限制就可以了。显然,随着移位
                 // 在每次选择列之前进行,原来N×N网格中某个已放置的皇后针对其对角线
                 // 上产生的限制都被记录下来了。 
 test(row + p, (ld + p) << 1, (rd + p) >> 1);                              
  }
   } 
   else    
   {
 // row的所有位都为1,即找到了一个成功的布局,回溯。
  sum++;
   }
}

int main(int argc, char *argv[])
{
   time_t tm;
   int n = 16;

   if (argc != 1)
  n = atoi(argv[1]);
   tm = time(0);

   // 因为整型数的限制,最大只能32位,
   // 如果想处理N大于32的皇后问题,需要
   // 用bitset数据结构进行存储。
   if ((n < 1) || (n > 32))                  
   {
  printf(" 只能计算1-32之间\n");
  exit(-1);
   }
   printf("%d 皇后\n", n);

   // N个皇后只需N位存储,N列中某列有皇后则对应bit置1。
   upperlim = (upperlim << n) - 1;          

   test(0, 0, 0);
   printf("共有%ld种排列, 计算时间%d秒 \n", sum, (int) (time(0) - tm));
}

上述代码容易看懂,但我觉得核心的是在针对试探-回溯算法所用的数据结构的设计上。
程序采用了递归,也就是借用了编译系统提供的自动回溯功能。

算法的核心:使用bit数组来代替以前由int或者bool数组来存储当前格子被占用或者说可用信息,从这

可以看出N个皇后对应需要N位表示。
巧妙之处在于:以前我们需要在一个N*N正方形的网格中挪动皇后来进行试探回溯,每走一步都要观察

和记录一个格子前后左右对角线上格子的信息;采用bit位进行信息存储的话,就可以只在一行格子也

就是(1行×N列)个格子中进行试探回溯即可,对角线上的限制被化归为列上的限制。
程序中主要需要下面三个bit数组,每位对应网格的一列,在C中就是取一个整形数的某部分连续位即可

。
row用来记录当前哪些列上的位置不可用,也就是哪些列被皇后占用,对应为1。
ld,rd同样也是记录当前哪些列位置不可用,但是不表示被皇后占用,而是表示会被已有皇后在对角线

上吃掉的位置。这三个位数组进行“或”操作后就是表示当前还有哪些位置可以放置新的皇后,对应0

的位置可放新的皇后。如下图所示的8皇后问题求解得第一步:
row:        [ ][ ][ ][ ][ ][ ][ ][*]
ld:         [ ][ ][ ][ ][ ][ ][*][ ]
rd:         [ ][ ][ ][ ][ ][ ][ ][ ]
------------------------------------
row|ld|rd:  [ ][ ][ ][ ][ ][ ][*][*]
所有下一个位置的试探过程都是通过位操作来实现的,这是借用了C语言的好处,详见代码注释。 

发布了10 篇原创文章 · 获赞 0 · 访问量 179

猜你喜欢

转载自blog.csdn.net/qq_45126968/article/details/105256246