Leetcode刷题Python155. 最小栈

Posted 2022-05-16 Better Bench

1 题目

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

实现 MinStack 类:

MinStack() 初始化堆栈对象。
void push(int val) 将元素val推入堆栈。
void pop() 删除堆栈顶部的元素。
int top() 获取堆栈顶部的元素。
int getMin() 获取堆栈中的最小元素。

2 解析

使用Python中的list作为栈，每次入栈的时候是一个元组（当前入站值，当前栈最小值）。当要返回但前栈最小值时，返回栈顶元组的第二个元素即可。

3 python实现

class MinStack:

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

    def push(self, x):
        if len(self.stack)==0:
            self.stack.append((x, x))
        else:
        # 将当前栈的最小值入栈
            self.stack.append((x, min(x, self.stack[-1][1])))

    def pop(self):
        self.stack.pop()

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

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