首页 > 代码库 > 【leetcode】Min Stack -- python版
【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 nothing14 def pop(self):15 p = self.stack.pop()16 #print ‘pop ‘ , p17 if p == self.minStack[-1]:18 #print ‘minn pop‘19 self.minStack.pop()20 21 # @return an integer22 def top(self):23 return self.stack[-1]24 25 # @return an integer26 def getMin(self):27 return self.minStack[-1]
AC
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 nothing17 def pop(self):18 p = self.stack.pop()19 #print ‘pop ‘ , p20 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 integer28 def top(self):29 return self.stack[-1]30 31 # @return an integer32 def getMin(self):33 #print self.minStack34 return self.minStack[-1][0]
【leetcode】Min Stack -- python版
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。