题: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()