八皇后问题是一个以国际象棋为背景的问题,如何在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的情况下)。 其中的一个结果数据类似(*表示皇后位置)如下所示:
*-------
----*---
-------*
-----*--
--*-----
------*-
-*------
---*----