Code
# -*- coding:utf-8 -*-
# 冲突检查
def conflict(state, nextY):
nextX = len(state)
# print(nextX),
for i in range(nextX): # 新位置要和之前的位置一一比较
if abs(state[i] - nextY) in (0, nextX - i):
# 纵坐标减去下一个皇后的纵坐标的绝对值 处于 0到下一皇后横坐标减i则冲突
return True # 冲突
return False # 不冲突
def queens(num, state=()):
# num = 8
# print("%d "%len(state)),
for pos in range(num): # 遍历每一行的单元格
if not conflict(state, pos): # 没有冲突
if len(state) == num - 1: # 递归基例
yield (pos,)
else:# 把没有冲突皇后的位置,添加到状态元组里,然后递归进层寻找合适的
for result in queens(num, state + (pos,)):
yield (pos,) + result
def prettyprint(solution):
def line(pos, length=len(solution)):
return '. ' * (pos) + 'X ' + '. ' * (length - pos - 1)
for pos in solution:
print(line(pos))
if __name__ == "__main__":
solutions = list(queens(4))
for i in solutions: # 打印所有的解法
print(i)
prettyprint(i)
print('*' * 100)
# prettyprint(random.choice(list(queens(8)))) # 可能有多种解法,展示其中一种解法
Q&A
Preview:
1.solutions中的每一个元素都是一种解法
2. 例如 (0, 4, 7, 5, 2, 6, 1, 3), 下标代表每一行,每一个元素代表当前皇后处于第几列
3. 如果下一个皇后的位置与当前的皇后位置相邻(包括上下,左右)或在同一对角线上,则说明有冲突,需要重新摆放
4.找到未冲突的皇后位置,深层递归,递归完毕后,得益于yield, 还可以从上一层的下一次位置继续深层次递归
5.for result in queens(num, state + (pos,)) 一方面是递归进层的入口,另一方面是递归退层的出口
- 宫格多大?依据是啥
- 答:N皇后代表N*N宫格
- nextX = len(state) 啥意思
- 答:例如state中有两个元素,代表nextX要去遍历第三行<=len(state)>中的元素
- 递归结束循环的条件<N皇后成立的标志>是啥
- 答 新位置和state中的元素冲突了,就会递归退层…往深层次说,当且仅当n = 1或n ≥ 4时问题有解
- queens 中的代码结构很凌乱,帮我梳理下
- 答:第一层for 主要控制遍历每一行第几个单元格; 第一个if 判断则是用来检查冲突,第二个if 用来判断递归结束条件,第一个else下的for循环主要用来深层次遍历
- yield 和return 的区别
- 答:前者是结束当前程序,然后还从停止的下一句语句继续运行, 后者是直接结束程序,只会深层递归一次,然后直接结束程序
- 递归开始退层的瞬间
- 答:
测试
将两个yield变成return, 报错 nonetype object isn’t iterable, 修改成只有一个return, 打印空白
MORE
相关文件:微云–N后问题
欢迎关注我的公众号:趣享Eureka<分享音乐,软件使用小技巧>