数据研发笔试Leetcode刷题笔记28：包含min函数的栈

最新推荐文章于 2024-07-22 11:51:43 发布

Lynn Wen

最新推荐文章于 2024-07-22 11:51:43 发布

阅读量197

点赞数

分类专栏：数据研发笔试Leetcode刷题笔记文章标签：数据结构 leetcode 算法 python

本文链接：https://blog.youkuaiyun.com/weixin_41961559/article/details/105964952

版权

数据研发笔试Leetcode刷题笔记专栏收录该内容

31 篇文章

订阅专栏

文章目录

1 题目描述
2 解题思路
3 代码实现（Python3）
4 复杂度分析

1 题目描述

来源：力扣（LeetCode）

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中，调用 min、push 及 pop 的时间复杂度都是 O(1)。
在这里插入图片描述

2 解题思路

在这里插入图片描述

3 代码实现（Python3）

class MinStack:
    def __init__(self):
        self.A, self.B = [], []

    def push(self, x: int) -> None:
        self.A.append(x)
        if not self.B or self.B[-1] >= x:
            self.B.append(x)

    def pop(self) -> None:
        if self.A.pop() == self.B[-1]:
            self.B.pop()

    def top(self) -> int:
        return self.A[-1]

    def min(self) -> int:
        return self.B[-1]