【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

 1 class MinStack:
 2     def __init__(self):
 3         self.stack = []
 4         self.minStack = []
 5     # @param x, an integer
 6     # @return an integer
 7     def push(self, x):
 8         self.stack.append(x)
 9         if len(self.minStack) == 0 or self.minStack[-1] >= x:
10             #print 'minn change'
11             self.minStack.append(x)
12  
13     # @return nothing
14     def pop(self):
15         p = self.stack.pop()
16         #print 'pop ' , p
17         if p == self.minStack[-1]:
18             #print 'minn pop'
19             self.minStack.pop()
20 
21     # @return an integer
22     def top(self):
23         return self.stack[-1]
24 
25     # @return an integer
26     def getMin(self):
27         return self.minStack[-1]

 1 class MinStack:
 2     def __init__(self):
 3         self.stack = []
 4         self.minStack = []
 5         #self.minStack.append(0)
 6     # @param x, an integer
 7     # @return an integer
 8     def push(self, x):
 9         self.stack.append(x)
10         if len(self.minStack) == 0 or self.minStack[-1][0] > x:
11             #print 'minn change'
12             self.minStack.append((x,1))
13         elif x == self.minStack[-1][0]:
14             self.minStack[-1] = (x, self.minStack[-1][1] + 1)
15  
16     # @return nothing
17     def pop(self):
18         p = self.stack.pop()
19         #print 'pop ' , p
20         if p == self.minStack[-1][0]:
21             if self.minStack[-1][1] > 1:
22                 #print 'minn pop'
23                 self.minStack[-1] = (self.minStack[-1][0], self.minStack[-1][1] - 1)
24             else:
25                 self.minStack.pop()
26 
27     # @return an integer
28     def top(self):
29         return self.stack[-1]
30 
31     # @return an integer
32     def getMin(self):
33         #print self.minStack
34         return self.minStack[-1][0]