八皇后问题回溯法动态演示2021_11_18版by李兴球

遥记我20岁左右的时候,拿着一本数据结构与算法书,PASCAL的,那个时候也是否对计算机编程充满了热爱和向往。时光飞速,我还是那个“少年”。仍旧兴趣盎然地在编程,想想我曾经是熟练的掌握了汇编语言的,可是现在我却看不懂了。写的八皇后算法那代码好糟糕。去年用我自己编写的精灵模块把八皇后代码重写了一遍,最近再次学习数据结构与算法,对动态规划、哈夫曼树、汉密尔顿路径等,感觉又有较大的长颈,确实体验到了温故而知新,收获不少。今天又把八皇后算法写了一遍,逻辑清晰了很多,对程序=数据结构+算法有了更深刻的体验。

Python八皇后演示动图

这个程序使用一个stack列表存储皇后的位置,当成栈使用。使用常见的回溯法,不断地去试探下一个位置,如果和前面的位置有冲突,则继续尝试。如果一行所有的列都尝试完了,再次回溯。代码中有注释,详情见代码:

 

import turtle
from eight_queen import *

def check_collision(stack):
    """检测皇后在row,col位置是否和其它皇后冲突"""
    row,col = stack[-1]                          # 最后摆放的  
    for x,y in stack[:-1]:
       if  abs(y-col) in (0,row-x) :return True # 列的差值等于0或者和最后一行到x行的差值一样
     
    return False           

def traceback(chess,):
    """从栈中弹出上次坐标,清除所盖图章"""
    i,j = stack.pop()
    chess[i][j] = 0     
    turtle.clearstamps(-1)  # 清除这个坐标的皇后
    j = j + 1               # 上一行的下一列位置
    return i,j
           
def output(stack):          # 以文本方式输出一个解
    for i,j in stack:
        print(j,end='')
    print()
           
chess = [[0,0,0,0,0,0,0,0],
         [0,0,0,0,0,0,0,0],
         [0,0,0,0,0,0,0,0],
         [0,0,0,0,0,0,0,0],
         [0,0,0,0,0,0,0,0],
         [0,0,0,0,0,0,0,0],
         [0,0,0,0,0,0,0,0],
         [0,0,0,0,0,0,0,0]]

turtle.setup(800,800)
turtle.speed(0)
draw_cross(770)                # 画坐标线
draw_grid(8,70)                # 画格子
turtle.penup()                 # 抬笔
turtle.addshape('queen.gif')
turtle.shape('queen.gif')
turtle.speed(1)
turtle.title('八皇后问题回溯法动态演示2021_11_18版by李兴球')
w = turtle.Turtle(visible=False)
w.penup()
w.sety(300)
ft = ('黑体',32,'normal')
w.write("八皇后问题回溯法动态演示",align='center',font=ft)

stack = []                    # 新建一个列表存储当前皇后坐标,当栈用
i ,j = 0,0

while True:
    if j<8:
       chess[i][j] = 1         # 放皇后
       place_queen(8,70,i,j)   # 在棋盘上放置皇后(盖图章)
       
    else:                      # 一行都尝试完了,超过最右边了,需要回溯
        if stack:              # 如果栈不是空的    
           i,j = traceback(chess) # 回溯到上一个位置的右边一列
           continue
        else:
           break
        
    stack.append((i,j))        # 放入stack中,保存以便回到上一个位置
    
    if check_collision(stack)  :# 发生冲突       
       i,j = traceback(chess)     # 和其它皇后发生冲突,需要回溯
       continue
    else:        
       i = i + 1               # 到下一行
       j = 0                   # 从0开始放皇后
       
       if i==8:           
           output(stack)       # 到了最后输出一个解
           i,j = traceback(chess) # 这当然也要继续到上一行继续试探
           xsleep(3)       
           
           
     

关于李兴球

李兴球的博客是Python创意编程原创博客
此条目发表在python, turtle分类目录,贴了标签。将固定链接加入收藏夹。

发表回复