欢迎来到山村网

python小程序----解迷宫图片

2019-03-02 11:20:53浏览:446 来源:山村网   
核心摘要:  一共做了三种,前两种分别是bfs和Astar,第三种不是解迷宫,是连通区域标记,用的是并查集。  效果如下(csdn貌似不能显示g

  一共做了三种,前两种分别是bfs和Astar,第三种不是解迷宫,是连通区域标记,用的是并查集。

  效果如下(csdn貌似不能显示gif动态图,所以只是截了几幅图片。

  本人觉得gif图 还是很好看的,github有:https://github.com/scauPYygzx/picture_maze ^-^):

python小程序----解迷宫图片    山村

bfs

Astar

并查集

  从图中可以看出Astar比bfs搜少了好多点,但是因为用到了priorityQueue的原因,在这图上的时间效率不如bfs。

  PS:用convert命令制作gif图真的很方便,推荐。

  code:

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 #-*- coding:utf-8 -*- from PIL import Image import Queue import time #只是玩玩就硬编码了.. ext = png imageName = p.%s % ext saveName = {0:05d}.{1} #用PIL库载入图片 im = Image.open(imageName) im = im.convert(RGB) print im.format, im.size, im.mode pix = im.load() size = im.size half = 256 >> 1 white = (255, 255, 255) black = (0, 0, 0) red = (255, 0, 0) green = (0, 255, 0) blue = (0, 0, 255) yellow = (255, 255, 0) fillColor = red lineColor = yellow colors = [red, green, blue, yellow, (0, 223, 248), (223, 8, 248), (255, 2, 146)] cLen = len(colors) #起点,终点 startPoint = [400, 984] endPoint = [398, 25] fa = [0] * size[0] * size[1] def changeImage(): ''' 将图片转换为二值图 ''' x, y = size for i in xrange(x): for j in xrange(y): r, g, b = pix[i, j] if r > half and g > half and b > half: pix[i, j] = white else: pix[i, j] = black #================================================================# #BFS def getNeighbor(point): x, y = point return [[x, y + 1], [x + 1, y], [x, y - 1], [x - 1, y]] def inBounds(point, size): x, y = point sx, sy = size if x < 0 or y < 0 or x >= sx or y >= sy: return False return True def isWhite(point): x, y = point if pix[x, y] == white: return True return False def bfs(start, end): count = no = 0 q = [[start]] while len(q): path = q.pop(0); p = path[-1] if p[0] == end[0] and p[1] == end[1]: im.save(saveName.format(no, ext)) no += 1 for pa in path: pix[pa[0], pa[1]] = pix[pa[0], pa[1] - 1] = pix[pa[0], pa[1] - 2] = lineColor count += 1 if not count % 200: im.save(saveName.format(no, ext)) no += 1 no += 1 im.save(saveName.format(no, ext)) return path for ne in getNeighbor(p): if inBounds(ne, size) and isWhite(ne): newPath = list(path) newPath.append(ne) q.append(newPath) pix[ne[0], ne[1]] = fillColor if not count % 10000: im.save(saveName.format(no, ext)) no += 1 count += 1 return None #================================================================# #Astar def getValue(point, end): ''' 启发函数 ''' return abs(point[0] - end[0]) + abs(point[1] - end[1]) def aStar(start, end): fillColor = green lineColor = black count = no = 0 Q = Queue.PriorityQueue() Q.put([0,[start]]) while(Q.qsize()): q = Q.get() path = q[-1] p = path[-1] if p[0] == end[0] and p[1] == end[1]: im.save(saveName.format(no, ext)) no += 1 for pa in path: pix[pa[0], pa[1]] = pix[pa[0], pa[1] - 1] = pix[pa[0], pa[1] - 2] = lineColor count += 1 if not count % 200: im.save(saveName.format(no, ext)) no += 1 im.save(saveName.format(no, ext)) return path for ne in getNeighbor(p): if inBounds(ne, size) and isWhite(ne): newPath = list(path) newPath.append(ne) newList = [getValue(p, end) + len(newPath), newPath] Q.put(newList) pix[ne[0], ne[1]] = fillColor if not count % 10000: im.save(saveName.format(no, ext)) no += 1 count += 1 return None #================================================================# #并查集 def getID(x, y): return x * size[0] + y def ufInit(size): for i in range(size[0]): for j in range(size[1]): fa[getID(i, j)] = getID(i, j) def ufFind(v): if v != fa[v]: fa[v] = ufFind(fa[v]) return fa[v] return fa[v] def ufUnion(x, y): x = ufFind(x) y = ufFind(y) if x == y: return fa[x] = y def connectivity(): fillColor = yellow no = count = now = 0 d = {} x_range = (5, 792) y_range = (22, 986) ufInit(size) for x in range(x_range[0], x_range[1]): for y in range(y_range[0], y_range[1]): if pix[x, y] != black: if pix[x - 1, y] == black and pix[x, y - 1] == black: pass elif pix[x - 1, y] == white and pix[x, y - 1] == black: ufUnion(getID(x - 1, y), getID(x, y)) elif pix[x - 1, y] == black and pix[x, y - 1] == white: ufUnion(getID(x, y - 1), getID(x, y)) else: ufUnion(getID(x - 1, y), getID(x, y - 1)) ufUnion(getID(x, y), getID(x - 1, y)) for x in range(x_range[0], x_range[1]): for y in range(y_range[0], y_range[1]): count += 1 if pix[x, y] != black: father = ufFind(getID(x,y)) if father not in d: d[father] = colors[now % cLen] now += 1 pix[x, y] = d[father] if not count % 15000: im.save(saveName.format(no, ext)) no += 1 im.save(saveName.format(no, ext)) #================================================================# t1 = time.time() changeImage() #bfs(startPoint, endPoint) #aStar(startPoint, endPoint) connectivity() t2 = time.time() print t2- t1
(责任编辑:豆豆)
下一篇:

ajax翻页效果模仿yii框架自己写的

上一篇:

nginx上的301重定向代码

  • 信息二维码

    手机看新闻

  • 分享到
打赏
免责声明
• 
本文仅代表作者个人观点,本站未对其内容进行核实,请读者仅做参考,如若文中涉及有违公德、触犯法律的内容,一经发现,立即删除,作者需自行承担相应责任。涉及到版权或其他问题,请及时联系我们 xfptx@outlook.com