LeetCode——最小栈

LeetCode——最小栈

题目描述：

设计一个支持 push ，pop ，top 操作，并能在常数时间内检索到最小元素的栈。

push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
getMin() —— 检索栈中的最小元素。

示例:

输入：
[“MinStack”,“push”,“push”,“push”,“getMin”,“pop”,“top”,“getMin”]
[[],[-2],[0],[-3],[],[],[],[]]

输出：
[null,null,null,null,-3,null,0,-2]

解释：
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.

提示：

pop、top 和 getMin 操作总是在 非空栈 上调用。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/min-stack
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路：

解法一：
我们用一个辅助栈来存储最小值。栈顶元素为当前的最小值。
push：如果栈一开始为空，我们把当前元素同时压入主栈和辅助栈；如果栈不为空，我们先把元素压入主栈，然后把该元素与辅助栈栈顶的元素（也就是当前最小值）进行比较，选出最小的那个压入辅助栈。
pop：同时删除主栈和辅助栈的栈顶元素。
top：返回主栈的栈顶元素。
getMin：返回辅助栈的栈顶元素。（为空返回None）

解法二：
我们定义一个新的类，命名为节点Node，将值（value）和最小值（min）封装在类中。
接下来的思路和刚才的解法一是一样的，不同的是不需要辅助栈，栈中存储的数据结构为节点。

这里提供两种解法的Python代码。

解法一Python代码：

class MinStack(object):

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

    def push(self, x):
        """
        :type x: int
        :rtype: None
        """
        self.stack.append(x)
        if self.min_stack:
            self.min_stack.append(min(x, self.min_stack[-1]))
        else:
            self.min_stack.append(x)

    def pop(self):
        """
        :rtype: None
        """
        self.stack.pop()
        self.min_stack.pop()

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

    def getMin(self):
        """
        :rtype: int
        """
        if self.min_stack is not None:
            return self.min_stack[-1]
        else:
            return None

# 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()

LeetCode测试结果：

解法二Python代码：

class MinStack(object):

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

    def push(self, x):
        """
        :type x: int
        :rtype: None
        """
        if self.stack:
            self.stack.append(Node(x, min(x, self.stack[-1].min)))
        else:
            self.stack.append(Node(x, x))

    def pop(self):
        """
        :rtype: None
        """
        self.stack.pop()

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

    def getMin(self):
        """
        :rtype: int
        """
        if self.stack is not None:
            return self.stack[-1].min
        else:
            return None


class Node(object):

    def __init__(self, value, min):
        self.value = value
        self.min = min

# 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()

LeetCode测试结果：