Min Stack [LeetCode 155]

1- 问题描述

Design a stack that supports push, pop, top, and retrieving the minimum element in constant time.

• push(x) -- Push element x onto stack.
• pop() -- Removes the element on top of the stack.
• top() -- Get the top element.
• getMin() -- Retrieve the minimum element in the stack

---

2- 思路分析

　　使用两个栈，一个栈保存原始数据(stack)，另一个栈保持每个元素对应的最小元素(minStack)。如stack中从栈底到栈顶为[3, 1, 5, 7, 0]，那么minStack中从栈底到栈顶为[3, 1, 1, 1, 0]。可参考[LeetCode] Min Stack in Python

---

3- Python实现

# -*- coding: utf-8 -*-
# Min Stack

class MinStack:
    def __init__(self):
        self.stack = []
        self.min = []    # 保存最小值

    # @param x, an integer
    # @return an integer
    def push(self, x):    # 入栈
        self.stack.append(x)
        if not self.min: 　　# 如果min栈为空则直接入栈
            self.min.append(x)
            return x
        if x < self.min[-1]:　　# 如果入栈元素x比min栈顶小，则x入栈
            self.min.append(x)
        else:　　# 如果x比min栈顶大，则重复栈顶元素
            self.min.append(self.min[-1])
        return x

    # @return nothing
    def pop(self):    #出栈
        self.stack.pop()
        self.min.pop()　　# min栈同步

    # @return an integer
    def top(self):    #栈顶
        return self.stack[-1]

    # @return an integer
    def getMin(self):    #栈内最小值
        return self.min[-1]