算法四:栈实现深度优先搜索--python
深度优先搜索
回溯法的思想,从一个节点开始,任意找下一个能走的点,当找到不能走的点时,退回到上一个点寻找是否有其他方向的点。使用栈存储当前路径。
代码简单,但是找到的路径并不一定是最短路径
深度优先实现迷宫问题
class Stack:
def __init__(self):
self.stack=[]
def push(self,element):
self.stack.append(element)
def pop(self):
return self.stack.pop()
def get_top(self):
if len(self.stack)>0:
return self.stack[-1]
else:
return None
#1表示迷宫的墙,0表示迷宫的路径
maze=[
[1,1,1,1,1,1,1,1,1,1],
[1,0,0,1,0,0,0,1,0,1],
[1,0,0,1,0,0,0,1,0,1],
[1,0,0,0,0,1,1,0,0,1],
[1,0,1,1,1,0,0,0,0,1],
[1,0,0,0,1,0,0,0,0,1],
[1,0,1,0,0,0,1,0,0,1],
[1,0,1,1,1,0,1,1,0,1],
[1,1,0,0,0,0,0,0,0,1],
[1,1,1,1,1,1,1,1,1,1],
]
dirs=[
lambda x,y:(x+1,y),
lambda x,y:(x-1,y),
lambda x,y:(x,y-1),
lambda x,y:(x,y+1),
]
def maze_path(x1,y1,x2,y2):#起始位置,终点位置
stack=[]
stack.append((x1,y1))
while(len(stack)>0):#栈空表示没有路了
curNode=stack[-1]
if curNode[0]==x2 and curNode[1]==y2:
#走到终点了,把路径打印出来
for p in stack:
print(p)
return True
for dir in dirs:
nextNode=dir(curNode[0],curNode[1])
#如果下一个节点能走
if maze[nextNode[0]][nextNode[1]]==0:
stack.append(nextNode)
maze[nextNode[0]][nextNode[1]]=2 #表示已经走过
break
else:#下个节点走不了,回溯
maze[nextNode[0]][nextNode[1]]=2
stack.pop()
else:
print("无路可走!")
return False
maze_path(1,1,8,8)