nowcoder 可查询最值的栈

最新推荐文章于 2022-08-29 08:00:00 发布

sparksnail

最新推荐文章于 2022-08-29 08:00:00 发布

阅读量222

点赞数

CC 4.0 BY-SA版权

分类专栏： OJ-nowcoder

本文链接：https://blog.youkuaiyun.com/y12345678904/article/details/79143857

OJ-nowcoder 专栏收录该内容

70 篇文章

订阅专栏

本文介绍了一种在栈数据结构中高效获取最小元素的方法。通过维护一个额外的辅助栈（minstack），每次压栈和弹栈时更新最小值，使得在任何时候都能快速访问到栈内的最小元素。

题目

定义栈的数据结构，请在该类型中实现一个能够得到栈最小元素的min函数。

思路

定义一个minstack，同时记录当前栈顶最小值。

代码

class Solution:
    def __init__(self):
        self.stack = []
        self.minstack = []
    def push(self, node):
        # write code here
        self.stack.append(node)
        if len(self.minstack) > 0:
            self.minstack.append(min(self.minstack[-1], node))
        else:
            self.minstack.append(node)
    def pop(self):
        # write code here
        self.stack.pop(-1)
        self.minstack.pop(-1)
    def top(self):
        # write code here
        return self.stack[-1]
    def min(self):
        # write code here
        return self.minstack[-1]