关于栈结构的一个例子1 借助栈结构求解迷宫问题

题目

给定一个（n+2）*（n+2）的二维数组
将数组最外一圈定义为1（代表墙）
将可以通过的路定义为0（代表路）
要求判断是否有结果，没有找到就输出错误

输入格式

在一行中输入迷宫maze出发点start和迷宫出口end，本题要求较低，出发点设置为（1,1），出口设置为（n，n）

输出格式

找得到就输出’Found’,找不到就输入’No way found’。

输入样例

a (1,1) (4,4)

输出样例

Found

思路

迷宫给定出口
首先我们可以定义探索迷宫的方式，将走过的地方推入栈结构，如果遇到了死路就将栈里面的数据推出栈，代表原路返回。
为了防止出现重复走死路的情况，所以我们要把走过的路定义为2，代表走过了。

代码如下

import numpy as np
def createMazeMatrix(n):#创建一个迷宫
    m=np.random.random((n+2,n+2))
    for i in range(0,n+2):
        for j in range(0,n+2):
            if m[i,j]>0.2:
                m[i,j]=0
            else:
                m[i,j]=1
    m[[0,-1],:]=np.ones((2,n+2))
    m[:,[0,-1]]=np.ones((n+2,2))
    m[1,1]=0
    m[n,n]=0
    return m
class Stack: # 定义了一个栈结构
    '''顺序存储 '''
    def __init__(self):  #创建一个空栈
        self.items=[]
    def isEmpty(self):   #判断栈是否为空
        return self.items==[]
    def push(self,item):  #元素进栈
        self.items.append(item)
    def pop(self):   #元素出栈
        return self.items.pop()
    def peek(self):  #查看最上面的元素
        return self.items[-1]
    def size(self):     #栈的大小
        return len(self.items)
def search(maze,start,end): # 搜索函数
    s = Stack()
    i,j = start[0],start[1]
    while i!=end[0] or j!=end[1]:
        if maze[i+1][j] == 0:#向下探索
            s.push([i,j])
            maze[i+1][j] = 2
            i+=1
        elif maze[i][j+1] == 0:#向右探索
            s.push([i,j])
            maze[i][j+1] = 2
            j+=1
        elif maze[i][j-1] == 0:#向左探索
            s.push([i,j-1])
            maze[i][j-1] = 2
            j-=1
        elif maze[i-1][j] == 0:#向上探索
            s.push([i-1,j])
            maze[i][j] = 2
            i-=1
        else:
            if not s.isEmpty():
                a = s.pop()
                i,j = a[0],a[1]
            else:
                return "No way found"
    plt.imshow(m,vmin=0,vmax=3)
    plt.show()
    return "Found"
print(search(m,(1,1),(20,20))) # 这是一个样例