[LeetCode] 155. Min Stack

题：https://leetcode.com/problems/min-stack/description/

题目

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.
Example:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); –> Returns -3.
minStack.pop();
minStack.top(); –> Returns 0.
minStack.getMin(); –> Returns -2.

思路

两种方法

1.

设两个 stack，一个作为数据stack，另一个存储数据min 的 stack。
当一个数据压入数据stack时，若该值 <= min stack 的top (其代表了 stack的最小值)，将其压入 min stack中。
当弹出一个数据时，若该值==min stack 的top，则min stack 也弹出。

2.

只设一个 stack 。
由于对于每个 min 可以视为 对 前面所有压入数据的统计。直到出现小于等于min的数据作为新已存入stack中数据的最小值。用min 可以 将 stack 分为多段，每个min表示前存入stack中数据最小值。
具体如下：
初始化，一个min值，该值为 flaot(‘inf’)。
当压入数据时，若该值比min值小，将min存入stack中，min更新为该数据；否则，不做处理。然后将该值存入stack中。
当弹出数据时。若弹出数据等于 min值。那么 stack中下个数据就是 之前存入数据的最小值。将这个数弹出用来更新 min。

code

########### 方法一 两个 stack ##########

class MinStack:

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

    def push(self, x):
        """
        :type x: int
        :rtype: void
        """
        self.dstack.append(x)
        if len(self.mstack) == 0 or x<=self.getMin():
            self.mstack.append(x)

    def pop(self):
        """
        :rtype: void
        """
        if self.dstack[-1] == self.mstack[-1]:
            del self.mstack[-1]
        del self.dstack[-1]


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


    def getMin(self):
        """
        :rtype: int
        """
        return self.mstack[-1]



# Your MinStack object will be instantiated and called as such:
# obj = MinStack()
# obj.push(x)
# obj.pop()
# param_3 = obj.top()
# param_4 = obj.getMin()

########### 方法二 一个stack ###########

class MinStack:

    def __init__(self):
        """
        initialize your data structure here.
        """
        self.dstack = []
        self.dmin = float('inf')
    def push(self, x):
        """
        :type x: int
        :rtype: void
        """
        if x <= self.dmin:
            self.dstack.append(self.dmin)
            self.dmin = x
        self.dstack.append(x)

    def pop(self):
        """
        :rtype: void
        """
        x = self.dstack.pop()
        if x == self.dmin:
            self.dmin = self.dstack.pop()



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


    def getMin(self):
        """
        :rtype: int
        """
        return self.dmin



# Your MinStack object will be instantiated and called as such:
# obj = MinStack()
# obj.push(x)
# obj.pop()
# param_3 = obj.top()
# param_4 = obj.getMin()