算法:回溯法与八皇后问题
在国际象棋中,皇后是最强大的一枚棋子,可以吃掉与其在同一行、列和斜线的敌方棋子。比中国象棋里的车强几百倍,比她那没用的老公更是强的飞起(国王只能前后左右斜线走一格)。上图右边高大的棋子即为皇后。
八皇后问题是这样一个问题:将八个皇后摆在一张8*8的国际象棋棋盘上,使每个皇后都无法吃掉别的皇后,一共有多少种摆法?此问题在1848年由棋手马克斯·贝瑟尔提出,岂止是有年头,简直就是有年头,82年的拉菲分分钟被秒的渣都不剩。
八皇后问题是典型的回溯法解决的问题,我们以这个问题为例介绍回溯法。
所谓回溯法,名字高大上,思想很朴素。设想把你放在一个迷宫里,想要走出迷宫,最直接的办法是什么呢?没错,试。先选一条路走起,走不通就往回退尝试别的路,走不通继续往回退,直到找到出口或所有路都试过走不出去为止。
是的你没说错,这就是暴力**。
尽管回溯法也算是暴力方法,但也不是特别暴力,特别暴力的相关部门都不让播,能播的都是可以接受的暴力。怎么说?考虑八皇后问题,解决这个问题最暴力的办法是这样的:
有关部门不让播的方法:
从8*8=64个格子里选8个格子,放皇后,测试是否满足条件,若满足则计数加1,否则换8个格子继续试。
很显然, 64中选8,并不是个小数字,十亿级别的尝试次数,够暴力。
这还是8*8的格子,要是换围棋棋盘……这画面太美我都不敢算。
稍加分析,我们可以得到一个不那么暴力的办法,显然,每行每列最多只能有一个皇后,如果基于这个事实进行暴力**,那结果会好得多。安排皇后时,第一行有8种选法,一旦第一行选定,假设选为(1,i),第二行只能选(2,j),其中j!=i,所以有7种选法。以此类推,需要穷举的情况有8!=40320种。
看起来这个结果已经不错了,但尝试的次数是随问题规模按阶乘水平提高的,BigMoyan仍然不满意——咋可能满意嘛!回溯法还没出,我要是满意了剩下的篇幅讲啥。
8皇后太多,后宫太过丰富BigMoyan可hold不住,不如我们先裁一半分析试试,于是BigMoyan与4位皇后办了离婚手续,同时把家缩小了一半,那么现在问题变成了4皇后问题,4个皇后在4*4的格子里各自安排不打架,一共有多少种安排方法?
试着来穷举一下,真的需要4!=24次尝试吗?
现在我们把第一个皇后放在第一个格子,被涂黑的地方是不能放皇后的。
第二行的皇后只能放在第三格或第四格,比方我们放第三格,则:
糟啦撸,前两位皇后沆瀣一气,已经把第三行全部锁死了,第三位皇后无论放哪里都难逃被吃掉的厄运。于是在第一个皇后位于1号,第二个皇后位于3号的情况下问题无解。我们只能返回上一步来,给2号皇后换个位置。
显然,第三个皇后只有一个位置可选。当第三个皇后占据第三行蓝色空位时,第四行皇后无路可走,于是发生错误,返回上层调用(3号皇后),而3号也别无可去,继续返回上层调用(2号),2号已然无路可去,继续返回上层(1号),于是1号皇后改变位置如下,继续搜索。
话说道这里,想必读者对“回溯法”已经有了基本概念。然而所谓知易行难,理解算法和将算法写出来完全是两回事。按照BigMoyan的风格,下面该进行算法分析了,下面的代码改写自刘汝佳《算法竞赛入门经典》,几乎是BigMoyan看到的实现8皇后问题最简洁的代码,改写后整个函数只有10行。
void queen(int row){
if(row==n)
total++;
else
for(int col=0;col!=n;col++){
c[row]=col;
if(is_ok(row))
queen(row+1);
}
}
算法是逐行安排皇后的,其参数row为现在正执行到第几行。n是皇后数,在八皇后问题里当然就是8啦。
第2行好理解,如果程序当前能正常执行到第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种解法。
全部程序如下:
#include<iostream>
#include<math.h>
using namespace std;
int n=8;
int total=0;
int *c=new int(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++;
else
for(int col=0;col!=n;col++){
c[row]=col;
if(is_ok(row))
queen(row+1);
}
}
int main(){
queen(0);
cout<<total;
return 1;
}
上一篇: 算法初识——回溯法八皇后
下一篇: PHP函数参数传递方法的具体改进技巧分享