算法学习-回溯法

 
八皇后问题是一个以国际象棋为背景的问题,如何在8*8的棋盘上放置8个皇后,使得任何一个皇后都无法直接吃掉其他的皇后。其实8皇后问题也可以推广为更为一般的n皇后问题,棋盘大小变为n*n,当n=2或者3时,是不存在解的,问题的限制有:
 
  • 所有的皇后都不能在同一行或同一列,也就是说每行或每列都只能存在一个皇后;
  • 所有的皇后都不能在对角线上,假设两个皇后的坐标为(i, j)和(k, l),明显当且仅当|i-k|=|j-l|时,两个皇后在一个对角线上。
 
回溯法是穷尽搜索算法的一种,采用试错的思想,尝试分步骤去解决一个问题,在分步解决问题的时候,当尝试不能得到有效的解答时,取消上一步或上几步的计算,通过其他可能的分步去得到可能的答案。
 
八皇后问题使用回溯法解决的思想是,假设某一行为当前状态,不断检查该行所有位置能否放置一个皇后:
 
  • 从一行中的第一个位置开始检查,如果不能放置,接着检查该行的第二个位置,依次检查下去,直到在该行中能够找到一个可以放置一个皇后的位置,然后保存该状态,转到下一行重复该步骤;
  • 如果检查了该行中的所有位置都不能放置一个皇后,说明上一行皇后放置的位置无法让所有的皇后找到自己的位置,需要回溯到上一行,继续将上一行中的皇后位置后移,直到找到合理的位置为止;
 
'''print current status'''
def print_current_status(queen):
    print "----new-result--"
    for i in range(0, max):
        line = ""
        for j in range(0, max):
            if queen[i] != j:
                line += "-"
            else:
                line += "*"
        print line

'''judge that whether the n position can place here'''
def can_place(n, queen):
    for i in range(0, n):
        if queen[i] == queen[n] or abs(queen[i] - queen[n]) == abs(i - n):
            return False
    return True

'''recursive execute n queen problem'''
def n_queen(n, queen):
    for i in range(0, max):
        queen[n] = i
        if can_place(n, queen):
            if n == max - 1:
                print_current_status(queen)
            else:
                n_queen(n + 1, queen)


if __name__ == '__main__':
    queen = [0 for i in range(max)]
    n_queen(0, queen)
 
 
其中核心的方法就是n_queen方法,方法接收的第一个参数就是当前已经遍历到了第n行,顺序地将第n行尝试0~max,如果期间有合适的,递归看一下n+1行是否可行;如果已经到了最后一行,就直接将数据结果输出,这就是其中的一个解;如果探索所有的可能解之后没有正确的结果,回溯至上一层,最多可能遍历8*8次(在max=8的情况下)。
 
 
其中的一个结果数据类似(*表示皇后位置)如下所示:
*-------
----*---
-------*
-----*--
--*-----
------*-
-*------
---*----
 

你可能感兴趣的:(算法,回溯)