剑指 Offer 30. 包含min函数的栈

最新推荐文章于 2024-08-23 14:01:41 发布

风去幽墨

最新推荐文章于 2024-08-23 14:01:41 发布

阅读量156

点赞数 1

分类专栏：剑指Offer（第2版）_python3 文章标签： leetcode python 剑指offer

本文链接：https://blog.youkuaiyun.com/wjl_zyl_1314/article/details/115869048

版权

剑指Offer（第2版）_python3 专栏收录该内容

57 篇文章

订阅专栏

本文介绍了一种特殊栈的数据结构实现，此栈包含获取最小元素的min函数，并确保min、push及pop操作的时间复杂度均为O(1)。通过使用两个列表分别存储栈元素及其对应的最小值，有效地实现了这一功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：

https://leetcode-cn.com/problems/bao-han-minhan-shu-de-zhan-lcof/

题意：

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

题解：

栈用python 里的list代替，出栈用pop（-1），进栈用append（x）
自定义的栈里再用另一个list来维护栈中最小值，其中mistack[i]表示栈中前i个元素中最小的元素值

代码：

class MinStack:

    def __init__(self):
        self.stack = []
        self.mistack = []


    def push(self, x: int) -> None:
        if len(self.mistack)>0:
            self.mistack.append(x if self.mistack[len(self.mistack)-1]>x else self.mistack[len(self.mistack)-1])
        else :
            self.mistack.append(x)
        self.stack.append(x)

    def pop(self) -> None:
        if len(self.stack)>0:
            self.stack.pop(-1)
            self.mistack.pop(-1)

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

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