Python Prim算法通过遍历墙实现迷宫的生成
之前，我们在另外一篇文章中使用Prim算法生成了一个完美迷宫，利用的是遍历网格的方法，这一次，我们要教教大家用遍历墙的方法生成，上一篇文章链接：Python利用Prim算法生成迷宫
我们需要用到随机库random，以及用来计算算法使用时间的time模块
导入这些模块
import random as rd import time
我们定义一个函数
def createMaze(a,b): # a:width b:height
添加一个变量储存算法开始的时间
startTime=time.time()
定义maze
maze={}
maze用来储存迷宫地图，格式如下：
{(n,"u"):0}
n表示第n个块，u d l r分别表示上下左右的墙壁，0表示没有墙壁，1表示有墙壁，初始是全部为1，生成的代码如下：
for n in range(a*b):
for face in ["u","d","l","r"]:
maze[(n,face)]=1
创建两个变量
history=[]
walls=[]
先初始随机选一个块并添加到遍历过的方块之中
block=rd.choice(list(maze.keys()))[0] history.append(block)
将这个方块的四个面的对应的墙都添加到候选墙的列表之中
for face in ["u","d","l","r"]:
walls.append((block,face))
只要候选墙不为空就一直循环
while len(walls)!=0:
选择一面墙，获取这个墙壁分割开来的两个块，如果已经到达边界外，则为None。注意，在最后一个elif之中，获取len(maze)要除以4，因为我们每个块有4个不同方向的墙壁，这个也是很容易疏忽的一点。
wall=rd.choice(walls)
twoBlocks=[wall[0]]
faces=[wall[1]]
if wall[1]=="u":
if wall[0]-a<0:
twoBlocks.append(None)
else:
twoBlocks.append(wall[0]-a)
faces.append("d")
elif wall[1]=="r":
if (wall[0]+1)%a!=0:
twoBlocks.append(wall[0]+1)
faces.append("l")
else:
twoBlocks.append(None)
elif wall[1]=="l":
if wall[0]%a!=0:
twoBlocks.append(wall[0]-1)
faces.append("r")
else:
twoBlocks.append(None)
elif wall[1]=="d":
if wall[0]+a>len(maze)/4-1:
twoBlocks.append(None)
else:
twoBlocks.append(wall[0]+a)
faces.append("u")
再定义两个列表
ins=[]
infaces=[]
获取这两个方块中有被添加到history的
for i,oneBlock in enumerate(twoBlocks):
if oneBlock in history:
ins.append(oneBlock)
infaces.append(faces[i])
因为只有一个被遍历过，所以我们就需要把这两个块中间的墙删掉，其实这里有两面，一面是第一个块的，另一个是第二个块相反方向的，只是重叠了，我们需要把这两面墙对应的值都设置为0，首先获取mirrorFace，也就是相反的方向，如果None在这两个方块的列表之中，那么就说明其中一个块在边上，所以就不需要再把这面墙删掉，保留这面墙，直接从候选墙之中删掉这面墙并开始新的循环，使用continue；如果他不是边上的块，也就是说twoBlocks里面没有None，就先把第一个块的那面墙去掉（改为0），然后获取另一个块放在other变量中，再把这第二个块的墙改为0，然后把这第二个块添加到history中，然后将这第二个块的四面墙都添加到候选墙中，注意，这里要添加的墙的值必须是1，也就是没有被检查遍历过的墙，如果候选墙已经有这面墙，就无需再添加，用for循环和if语句搭配，就可以简简单单写出这段代码，逻辑理清楚就不难写啦！代码如下：
if len(ins)==1:
mirrorFace=None
if infaces[0]=="u":
mirrorFace="d"
elif infaces[0]=="d":
mirrorFace="u"
elif infaces[0]=="r":
mirrorFace="l"
elif infaces[0]=="l":
mirrorFace="r"
if not (None in twoBlocks):
maze[(ins[0],infaces[0])]=0
other=None
if ins[0]==twoBlocks[0]:
other=twoBlocks[1]
else:
other=twoBlocks[0]
maze[(other,mirrorFace)]=0
walls.remove(wall)
history.append(other)
for face in ["u","l","r","d"]:
if maze.get((other,face))==1 and not ((other,face) in walls):
walls.append((other,face))
else:
walls.remove(wall)
continue
elif len(ins)==2:
walls.remove(wall)
写到这儿，我们的算法就差不多结束了，最后添加endTime获取算法结束时间
endTime=time.time()
并将它输出到控制台
print(f"生成迷宫使用时间：{endTime-startTime}秒")
返回迷宫
return maze
这个算法速度挺快的，99x99的迷宫只用了三秒多，一般三十多乘三十多的也只生成了30毫秒，效率很高！
到此这篇关于Python Prim算法通过遍历墙实现迷宫的生成的文章就介绍到这了,更多相关Python Prim生成迷宫内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要详解为大家介绍了Python如何将多张照片制作成视频，文章通过一个例子给大家介绍多张照片制作成视频的方法，代码示例介绍的非常详细，感兴趣的同学可以参考一下2023-07-07
这篇文章主要介绍了浅谈keras中的目标函数和优化函数MSE用法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-06-06
这篇文章主要介绍了Python连接数据库进行数据查询的操作代码,本文给大家介绍的非常详细,感兴趣的朋友跟随小编一起看看吧2024-06-06
今天这篇文章给大家介绍Python中的文件操作，文章内容介绍的很详细，有需要的可以参考借鉴。2016-08-08
这篇文章主要介绍了numpy在python中的用法,NumPy是Python科学计算库,主要用于处理大型多维数组和矩阵运算,它提供了多种函数进行数组操作,文中通过代码介绍的非常详细,需要的朋友可以参考下2025-01-01
这篇文章主要介绍了在Python程序中实现分布式进程的教程,在多进程编程中十分有用,示例代码基于Python2.x版本,需要的朋友可以参考下2015-04-04
这篇文章主要介绍了python如何给内存和cpu使用量设置限制，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-05-05
这篇文章主要为大家介绍了一个开源项目（Barbershop），可以将照片中的发型更换成另一个，文中实现过程讲解详细，感兴趣的可以学习一下2022-01-01
这篇文章主要介绍了Python2升级/安装pip报错问题及解决方案,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2024-03-03
这篇文章主要为大家详细介绍了Python如何利用Pygame实现接小弹珠游戏，即用挡板接住会反弹的小球，随着次数的增多，速度变快，分数增多，感兴趣的可以了解一下2022-12-12
最新评论