八皇后问题之python解法

八皇后的问题解析请见:

八皇后问题

这里给出Python的解法,相对于C语言,Python比较简单,代码量较小,但是效率也只能呵呵了,貌似很难两全啊。

首先定义冲突函数:

1
2
3
4
5
6
def conflict(state,nextX):
nextY = len(state)
for i in range(nextY):
if abs(state[i] - nextX) in (0,nextY - i):
return True
return False

递归:

1
2
3
4
5
6
7
8
9
10
def queen(num = 8,state = ()):
if len(state) == num-1:
for pos in range(num):
if not conflict(state,pos):
yield (pos,)
else:
for pos in range(num):
if not conflict(state,pos):
for result in queen(num,state+(pos,)):
yield (pos,)+result

对输出进行形象化处理:

1
2
3
4
5
def prettyprint(solution):
def line(pos,length = len(solution)):
return '. '*pos + '* ' + '. '*(length - pos - 1)
for pos in solution:
print(line(pos))

显示 Gitment 评论