Python LeetCode-155. 最小栈(难度-简单)(python)

1. 题目描述

设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。

  • push(x) -- 将元素 x 推入栈中。
  • pop() -- 删除栈顶的元素。
  • top() -- 获取栈顶元素。
  • getMin() -- 检索栈中的最小元素。

示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.

2.思路

  • 首先想到是利用python中的列表来做,进栈用append,出站用pop,但是这里还有一个小要求是要能检索到栈中的最小元素并且是在常数时间内。
  • 算法一般都是空间换时间或者时间换空间,要想得到最小元素并且是在常数时间内,在建栈的时候就保存这个索引就行了,详情见代码。

3.解决

class MinStack(object):

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

    def push(self, x):
        """
        :type x: int
        :rtype: void
        """
        current_min = self.getMin()
        if not current_min or x < current_min:
            current_min = x
        self.my_stack.append((x, current_min))  # 关键点在这,这里相当于保存每个元素所对应最小值

    def pop(self):
        """
        :rtype: void
        """
        self.my_stack.pop()

    def top(self):
        """
        :rtype: int
        """
        return self.my_stack[-1][0]

    def getMin(self):
        """
        :rtype: int
        """
        if len(self.my_stack) == 0:
            return None
        else:
            return self.my_stack[-1][1]


最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 题目描述 [最小栈] 设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。 push...
    一只可爱的柠檬树阅读 103评论 0 1
  • 传送门 题目要求 设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。 push(x...
    慧鑫coming阅读 292评论 0 0
  • 最小栈 题目 设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。 push(x) ...
    饮酒醉回忆阅读 202评论 0 1
  • 设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。 push(x) -- 将元素 ...
    小白学编程阅读 1,485评论 2 1
  • 题目描述 设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。 push(x) --...
    云胡同学阅读 311评论 0 0