专栏名称: 编程派
Python程序员都在看的公众号,跟着编程派一起学习Python,看最新国外教程和资源!
目录
相关文章推荐
Python开发者  ·  24 小时不眠不休的 AI 网红,年入 ... ·  3 天前  
Python开发者  ·  DeepSeek 四连炸!梁文峰参与开发 ·  3 天前  
Python爱好者社区  ·  王炸!DeepSeek彻底爆了! ·  3 天前  
Python爱好者社区  ·  在银行做开发半年,已经丧失跳槽能力。。。 ·  3 天前  
51好读  ›  专栏  ›  编程派

「Python 算法实战」:栈

编程派  · 公众号  · Python  · 2017-04-27 11:38

正文

作者:安生

原文:https://blog.ansheng.me/article/python-algorithm-combat-series-stack/

栈(stack)又称之为堆栈是一个特殊的有序表,其插入和删除操作都在栈顶进行操作,并且按照先进后出,后进先出的规则进行运作。


如下图所示

例如枪的弹匣,第一颗放进弹匣的子弹反而在发射出去的时候是最后一个,而最后放入弹匣的一颗子弹在打出去的时候是第一颗发射出去的。

栈的接口

如果你创建了一个栈,那么那么应该具有以下接口来进行对栈的操作

接口 描述
push() 入栈
pop() 出栈
isEmpty() 判断是否为空栈
length() 获取栈的长度
getTop() 取栈顶的元素,元素不出栈

知道栈需要上述的接口后,那么在Python中,列表就类似是一个栈,提供接口如下:

操作 描述
s = [] 创建一个栈
s.append(x) 往栈内添加一个元素
s.pop() 在栈内删除一个元素
not s 判断是否为空栈
len(s) 获取栈内元素的数量
s[-1] 获取栈顶的元素

Python中的栈接口使用实例:

  1. # 创建一个栈

  2. In [1]: s = []

  3. # 往栈内添加一个元素

  4. In [2]: s.append(1)

  5. In [3]: s

  6. Out[3]: [1]

  7. # 删除栈内的一个元素

  8. In [4]: s.pop()

  9. Out[4]: 1

  10. In [5]: s

  11. Out[5]: []

  12. # 判断栈是否为空

  13. In [6]: not s

  14. Out[6]: True

  15. In [7]: s.append(1)

  16. In [8]: not s

  17. Out[8]: False

  18. # 获取栈内元素的数量

  19. In [9]: len(s)

  20. Out [9]: 1

  21. In [10]: s.append(2)

  22. In [11]: s.append(3)

  23. # 取栈顶的元素

  24. In [12]: s[-1]

  25. Out[12]: 3

一大波实例

在了解栈的基本概念之后,让我们再来看几个实例,以便于理解栈。

括号匹配

题目

假如表达式中允许包含三中括号 () [] {} ,其嵌套顺序是任意的,例如:

正确的格式

  1. {()[()]},[{({})}]

错误的格式

  1. [(]),[()),(()}

编写一个函数,判断一个表达式字符串,括号匹配是否正确

思路

  1. 创建一个空栈,用来存储尚未找到的左括号;

  2. 便利字符串,遇到左括号则压栈,遇到右括号则出栈一个左括号进行匹配;

  3. 在第二步骤过程中,如果空栈情况下遇到右括号,说明缺少左括号,不匹配;

  4. 在第二步骤遍历结束时,栈不为空,说明缺少右括号,不匹配;

解决代码

建议在 pycharm 中打断点,以便于更好的理解

  1. #!/use/bin/env python

  2. # _*_ coding:utf-8 _*_

  3. LEFT = {'(', '[', '{'}  # 左括号

  4. RIGHT = {')', ']', '}'}  # 右括号

  5. def match(expr):

  6.    """

  7.    :param expr:  传过来的字符串

  8.    :return:  返回是否是正确的

  9.    """

  10.    stack = []  # 创建一个栈

  11.    for brackets in expr:  # 迭代传过来的所有字符串

  12.        if brackets in LEFT:  # 如果当前字符在左括号内

  13.            stack.append(brackets)  # 把当前左括号入栈

  14.        elif brackets in RIGHT:  # 如果是右括号

  15.             if not stack or not 1 <= ord(brackets) - ord(stack[-1]) <= 2:

  16.                # 如果当前栈为空,()]

  17.                # 如果右括号减去左括号的值不是小于等于2大于等于1

  18.                return False  # 返回False

  19.            stack.pop()  # 删除左括号

  20.     return not stack  # 如果栈内没有值则返回True,否则返回False

  21. result = match('[(){()}]')

  22. print(result)

迷宫问题

题目

用一个二维数组表示一个简单的迷宫,用0表示通路,用1表示阻断,老鼠在每个点上可以移动相邻的东南西北四个点,设计一个算法,模拟老鼠走迷宫,找到从入口到出口的一条路径。

如图所示

出去的正确线路如图中的红线所示

思路

  1. 用一个栈来记录老鼠从入口到出口的路径

  2. 走到某点后,将该点左边压栈,并把该点值置为1,表示走过了;

  3. 从临近的四个点中可到达的点中任意选取一个,走到该点;

  4. 如果在到达某点后临近的4个点都不走,说明已经走入死胡同,此时退栈,退回一步尝试其他点;

  5. 反复执行第二、三、四步骤直到找到出口;

解决代码

  1. #!/use/bin/env python

  2. # _*_ coding:utf-8 _*_

  3. def initMaze():

  4.    """

  5.    :return: 初始化迷宫

  6.    """

  7.    maze = [[0] * 7 for _ in range(5 + 2)]  # 用列表解析创建一个7*7的二维数组,为了确保迷宫四周都是墙

  8.    walls = [  # 记录了墙的位置

  9.        (1, 3),

  10.        (2, 1), (2, 5),

  11.         (3, 3), (3, 4),

  12.        (4, 2),  # (4, 3),  # 如果把(4, 3)点也设置为墙,那么整个迷宫是走不出去的,所以会返回一个空列表

  13.        (5, 4)

  14.    ]

  15.     for i in range(7):  # 把迷宫的四周设置成墙

  16.        maze[i][0] = maze[i][-1] = 1

  17.        maze[0][i] = maze[-1][i] = 1

  18.    for i, j in walls:  # 把所有墙的点设置为1

  19.        maze[i][j] = 1

  20.    return maze

  21. """

  22. [1, 1, 1, 1, 1, 1, 1]

  23. [1, 0, 0, 1, 0, 0, 1]

  24. [1, 1, 0, 0, 0, 1, 1]

  25. [1, 0, 0, 1, 1, 0, 1]

  26. [1, 0, 1, 0, 0, 0, 1]

  27. [1, 0, 0, 0, 1, 0, 1]

  28. [1, 1, 1, 1, 1, 1, 1]

  29. """

  30. def path(maze, start, end):

  31.    """

  32.    :param maze: 迷宫

  33.    :param start: 起始点

  34.    :param end: 结束点

  35.    :return: 行走的每个点

  36.    """

  37.    i, j = start  # 分解起始点的坐标

  38.    ei , ej = end  # 分解结束点的左边

  39.    stack = [(i, j)]  # 创建一个栈,并让老鼠站到起始点的位置

  40.    maze[i][j] = 1  # 走过的路置为1

  41.     while stack:  # 栈不为空的时候继续走,否则退出

  42.        i, j = stack[-1]  # 获取当前老鼠所站的位置点

  43.        if (i, j) == (ei, ej): break  # 如果老鼠找到了出口

  44.         for di, dj in [(0, -1), (0, 1), (-1, 0), (1, 0)]:  # 左右上下

  45.            if maze[i + di][j + dj] == 0:  # 如果当前点可走

  46.                maze [i + di][j + dj] = 1  # 把当前点置为1

  47.                stack.append((i + di, j + dj))  # 把当前的位置添加到栈里面

  48.                break

  49.         else:  # 如果所有的点都不可走

  50.            stack.pop()  # 退回上一步

  51.    return stack  # 如果迷宫不能走则返回空栈

  52. Maze = initMaze()  # 初始化迷宫

  53. result = path(maze=Maze, start=(1, 1), end=(5, 5))  # 老鼠开始走迷宫

  54. print(result)

  55. # [(1, 1), (1, 2), (2, 2), (3, 2), (3, 1), (4, 1), (5, 1), (5, 2), (5, 3), (4, 3), (4, 4), (4, 5), (5, 5)]

后缀表达式求值

题目

计算一个表达式时,编译器通常使用后缀表达式,这种表达式不需要括号:

中缀表达式 后缀表达式
2 + 3 * 4 2 3 4 * +
( 1 + 2 ) * ( 6 / 3 ) + 2 1 2 + 6 3 / * 2 +
18 / ( 3 * ( 1 + 2 ) ) 18 3 1 2 + * /

编写程序实现后缀表达式求值函数。

思路

  1. 建立一个栈来存储待计算的操作数;

  2. 遍历字符串,遇到操作数则压入栈中,遇到操作符号则出栈操作数(n次),进行相应的计算,计算结果是新的操作数压回栈中,等待计算

  3. 按上述过程,遍历完整个表达式,栈中只剩下最终结果;

解决代码

  1. #!/use/bin/env python

  2. # _*_ coding:utf-8 _*_

  3. operators = {  # 运算符操作表

  4.     '+': lambda op1, op2: op1 + op2,

  5.    '-': lambda op1, op2: op1 - op2,

  6.    '*': lambda op1, op2: op1 * op2,

  7.     '/': lambda op1, op2: op1 / op2,

  8. }

  9. def evalPostfix(e):

  10.     """

  11.    :param e: 后缀表达式

  12.    :return: 正常情况下栈内的第一个元素就是计算好之后的值

  13.    """

  14.    tokens







请到「今天看啥」查看全文