首页 > 其他 > 详细

剑指 Offer 30. 包含min函数的栈(辅助栈)

时间:2020-08-11 16:07:44      阅读:54      评论:0      收藏:0      [点我收藏+]
  • 题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。


示例:

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min();   --> 返回 -3.
minStack.pop();
minStack.top();      --> 返回 0.
minStack.min();   --> 返回 -2.
  • 题解

push,pop,top都很好做到O(1),但是min的话不容易。这里可以考虑用辅助栈,一个栈正常push数据,另一个栈的栈顶始终保存当前最小的数。

class MinStack:

    def __init__(self):
        """
        initialize your data structure here.
        """
        self.items = []
        self.assItems = []

    def push(self, x: int) -> None:
        self.items.append(x)
        if not self.assItems:
            self.assItems.append(x)
        else:
            if self.assItems[-1] > x:
                self.assItems.append(x)
            else:
                self.assItems.append(self.assItems[-1])

    def pop(self) -> None:
        self.items.pop()
        self.assItems.pop()

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

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

 

剑指 Offer 30. 包含min函数的栈(辅助栈)

原文:https://www.cnblogs.com/yeshengCqupt/p/13475796.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!