美文网首页互联网科技数据结构和算法分析程序员
小朋友学经典算法(14):回溯法和八皇后问题

小朋友学经典算法(14):回溯法和八皇后问题

作者: 海天一树X | 来源:发表于2018-10-09 17:02 被阅读75次

    一、回溯法

    回溯法(探索与回溯法)是一种选优搜索法,又称为试探法,按选优条件向前搜索,以达到目标。但当探索到某一步时,发现原先选择并不优或达不到目标,就退回一步重新选择,这种走不通就退回再走的技术为回溯法,而满足回溯条件的某个状态的点称为“回溯点”。

    二、八皇后问题

    (一)问题描述

    1.png

    在国际象棋中,皇后是最强大的一枚棋子,可以吃掉与其在同一行、列和斜线的敌方棋子。比中国象棋里的车强几百倍,比她那没用的老公更是强的飞起(国王只能前后左右斜线走一格)。
    八皇后问题是这样一个问题:将八个皇后摆在一张8*8的国际象棋棋盘上,使每个皇后都无法吃掉别的皇后,一共有多少种摆法?
    八皇后问题,是一个古老而著名的问题,是回溯算法的典型案例。该问题是国际西洋棋棋手马克斯·贝瑟尔于1848年提出。高斯认为有76种方案。1854年在柏林的象棋杂志上不同的作者发表了40种不同的解,后来有人用图论的方法解出92种结果。计算机发明后,有多种计算机语言可以解决此问题。

    (二)分析过程

    为了使问题简化,假定国王与四位皇后离了婚,那么只剩下四位皇后了。八皇后问题就变成了四皇后问题。

    2.png

    在第一行放1号皇后。第一行的四个格子都可以放。按枚举的习惯,先放在第一个格子。如下图所示。黑色的格子不能放其他的皇后。

    3.png

    在第二行放2号皇后,只能放在第三个或第四个格子。按枚举的习惯,先放在第三个格子,如下图所示。

    4.png

    不好了,前两位皇后沆瀣一气,已经把第三行全部锁死了,第三位皇后无论放哪里都难逃被吃掉的厄运。于是在第一个皇后位于1号,第二个皇后位于3号的情况下问题无解。我们只能返回上一步来,给2号皇后换个位置,挪到第四个格子上。

    5.png

    显然,第三个皇后只有一个位置可选。当第三个皇后占据第三行蓝色空位时,第四行皇后无路可走,于是发生错误,返回上层挪动3号皇后,而3号也别无可去,继续返回上层挪动2号皇后,2号已然无路可去,继续返回上层挪动1号皇后。于是1号皇后改变位置如下,继续搜索。

    6.png

    分析到这里,想必小朋友们对“回溯法”已经有了基本概念。下面要将算法实现出来。

    (三)代码实现

    01 void queen(int row)
    02 {
    03     if(row==n)
    04         total++;
    05     else
    06         for(int col=0;col!=n;col++)
    07         {
    08             c[row]=col;
    09             if(is_ok(row))
    10                 queen(row+1);
    11         }        
    12 }
    

    算法是逐行安排皇后的,其参数row为现在正执行到第几行。n是皇后数,在八皇后问题里当然就是8啦。
    第4行好理解,如果程序当前能正常执行到第8行,那自然是找到了一种解法,于是八皇后问题解法数加1。
      如果当前还没排到第八行,则进入else语句。遍历所有列col,将当前col存储在数组c里,然后使用is_ok()检查row行col列能不能摆皇后,若能摆皇后,则递归调用queen去安排下一列摆皇后的问题。

    还不太清楚?再慢点来,刚开始的时候row=0,意思是要对第0行摆皇后了。
      If判断失败,进入else,进入for循环,col初始化为0
      显然,0行0列的位置一定可以摆皇后的,因为这是第一个皇后啊,后宫空荡她想怎么折腾就怎么折腾,于是is_ok(0)测试成功,递归调用queen(1)安排第1行的皇后问题。
      第1行时row=1,进来if依然测试失败,进入for循环,col初始化为0。1行0列显然是不能摆皇后的,因为0行0列已经有一个圣母皇太后在那搁着了,于是is_ok()测试失败,循环什么也不做空转一圈,col变为1。1行1列依然is_ok()测试失败,一直到1行2列,发现可以摆皇后,于是继续递归queen(2)去安排第二个皇后位置。

    如果在某种情况下问题无解呢?例如前面在4皇后问题中,0行0列摆皇后是无解的。假设前面递归到queen(2)时候,发现第2行没有地方可以摆皇后,那怎么办呢?要注意queen(2)的调用是在queen(1)的for循环框架内的,queen(2)若无解,则自然而然queen(1)的for循环col自加1,即将第1行的皇后从1行2列改为1行3列的位置,检查可否放皇后后继续安排下一行的皇后。如此递归,当queen(0)的col自加到7,说明第一列的皇后已经遍历了从0行1列到0行7列,此时for循环结束,程序退出。

    在主函数中调用queen(0),得到正确结果,8皇后问题一共有92种解法。

    这里再看一下is_ok函数:

    bool is_ok(int row)
    {
    for(int j=0;j!=row;j++)
    {
            if(c[row]==c[j] || row-c[row]==j-c[j] || row+c[row]==j+c[j])
                return false;
        }
        return true;
    }
    

    这里row表示当前的行。假定当前的行为第3行(从0开始计数)。那么for循环里,j = 0表示第0行,j = 1表示第1行,j = 2表示第2行。
    c[row]表示第row行所在的列。比如c[3] = 2表示第三行第2列。c[0] = 2表示第0行第2列等。
    c[row] == c[j]表示第row行和第j行(j < row)的列一样,这样两个皇后就冲突了,所以返回false。
    row - c[row] == j - c[j],表示row行c[row]列与j行c[j]列,在同一条斜率为负的斜线上。这样两个皇后也冲突了。以下图为例

    7.png

    例1:
    A格子,j = 0, c[j] = c[0] = 0,即第0行第0列。C格子,row = 2, c[row] = c[2] = 2, 即第2行第2列。row - c[row] == j - c[j],表示这两个格子在一条斜线上,返回false。
    例2:
    B格子,j = 1, c[j] = c[1] = 0,即第1行第0列。D格子,row = 3, c[row] = c[3] = 2, 即第3行第2列。row - c[row] == j - c[j],表示这两个格子在一条斜线上,返回false。

    row+c[row]==j+c[j],表示row行c[row]列与j行c[j]列,在同一条斜率为正的斜线上。这样两个皇后也冲突了。如下图所示:

    8.png

    例3:
    A格子,j = 0, c[j] = c[0] = 1,即第0行第1列。B格子,row = 1, c[row] = c[1] = 0, 即第1行第0列。row + c[row] == j + c[j],表示这两个格子在一条斜线上,返回false。
    例4:
    C格子,j = 0, c[j] = c[0] = 3,即第0行第3列。D格子,row = 3, c[row] = c[3] = 0, 即第3行第0列。row + c[row] == j + c[j],表示这两个格子在一条斜线上,返回false。

    上面表示两种斜线的情况,一种用的是“-”,另一种用的是“+”,其实是因为这两种线的斜率分别为-1和1的缘故。

    完整代码如下:

    # #include<iostream>
    #include<math.h>
    using namespace std;
    
    int n=8;
    int total=0;
    int *c=new int(n); // 也可以写为int c[n];
    
    bool is_ok(int row)
    {
        for(int j=0;j!=row;j++)
        {
            if(c[row]==c[j] || row-c[row]==j-c[j] || row+c[row]==j+c[j])
                return false;
        }
    
        return true;
    }
    
    void queen(int row)
    {
        if(row==n)
        {
            total++;
            
            // 打印出8个皇后具体放在0~7行的第几列
            for(int i=0; i<n; i++)
                cout<<c[i]<<" ";
            cout<<endl;
        }
        else
        {
            for(int col=0;col!=n;col++)
            {
                c[row]=col;
                if(is_ok(row))
                    queen(row+1);
            }
        }
    }
    
    int main()
    {
        queen(0);
        cout << total << endl;
    
        return 0;
    }
    

    运行结果:

    ……
    6 2 0 5 7 4 1 3
    6 2 7 1 4 0 5 3
    6 3 1 4 7 0 2 5
    6 3 1 7 5 0 2 4
    6 4 2 0 5 7 1 3
    7 1 3 0 6 4 2 5
    7 1 4 2 0 6 3 5
    7 2 0 5 1 4 6 3
    7 3 0 2 5 1 6 4
    92
    

    三、回溯法和枚举法的区别

    回溯法与穷举法有某些联系,它们都是基于试探的。
    穷举法要将一个解的各个部分全部生成后,才检查是否满足条件,若不满足,则直接放弃该完整解,然后再尝试另一个可能的完整解,它并没有沿着一个可能的完整解的各个部分逐步回退生成解的过程。
    而对于回溯法,一个解的各个部分是逐步生成的,当发现当前生成的某部分不满足约束条件时,就放弃该步所做的工作,退到上一步进行新的尝试,而不是放弃整个解重来。

    了解小朋友学编程请加QQ307591841(微信与QQ同号)或QQ群581357582
    关注公众号请扫描二维码


    qrcode_for_kidscode_258.jpg

    相关文章

      网友评论

        本文标题:小朋友学经典算法(14):回溯法和八皇后问题

        本文链接:https://www.haomeiwen.com/subject/zhlyaftx.html