【leetcode】Min Stack -- python版

题目描述：

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.

解题思路：

这个问题挺简单（确实leetcode给的评级也是easy），但是还真是遇上了不少的问题。
首先，这道题目我们的想法是如何去做到这个特殊的getMin，想到的方法当然是空间换时间啦，那么用什么空间呢？当然是另一个栈啦，所以我们就有了这么一个想法：用另一个栈来记录当前最小值，那么查找最小值就不需要遍历了，这样就实现了时间复杂度和空间复杂度都是O(n)。的确这个想法已经很不错了，用java（官方题解就是这个版本）和C++（亲测）。但是我用python就MLE了，让我纠结了好久。
所以在没办法就要优化内存了，这里采用的方法是在minStack中插值的时候对相同的值不重复插入，而是记录他的次数，终于AC

MLE

class MinStack:
    def __init__(self):
        self.stack = []
        self.minStack = []
    # @param x, an integer
    # @return an integer
    def push(self, x):
        self.stack.append(x)
        if len(self.minStack) == 0 or self.minStack[-1] >= x:
            #print 'minn change'
            self.minStack.append(x)
 
    # @return nothing
    def pop(self):
        p = self.stack.pop()
        #print 'pop ' , p
        if p == self.minStack[-1]:
            #print 'minn pop'
            self.minStack.pop()

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

    # @return an integer
    def getMin(self):
        return self.minStack[-1]

AC

class MinStack:
    def __init__(self):
        self.stack = []
        self.minStack = []
        #self.minStack.append(0)
    # @param x, an integer
    # @return an integer
    def push(self, x):
        self.stack.append(x)
        if len(self.minStack) == 0 or self.minStack[-1][0] > x:
            #print 'minn change'
            self.minStack.append((x,1))
        elif x == self.minStack[-1][0]:
            self.minStack[-1] = (x, self.minStack[-1][1] + 1)
 
    # @return nothing
    def pop(self):
        p = self.stack.pop()
        #print 'pop ' , p
        if p == self.minStack[-1][0]:
            if self.minStack[-1][1] > 1:
                #print 'minn pop'
                self.minStack[-1] = (self.minStack[-1][0], self.minStack[-1][1] - 1)
            else:
                self.minStack.pop()

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

    # @return an integer
    def getMin(self):
        #print self.minStack
        return self.minStack[-1][0]