剑指offer-20.包含min函数的栈-Python

20.包含min函数的栈

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。
注意：保证测试中不会当栈为空的时候，对栈调用pop()或者min()或者top()方法。

记录

一些概念：
stack()：创建一个新的空栈，python里面使用列表表示：[]；
push(node)：添加一个新的元素node到栈顶，python里使用append；
pop()：弹出栈顶元素，python里使用pop()；
top()：返回最后压入栈的元素

方法一：
定义两个栈，一个是原栈stack，一个是专门用来存放最小值的栈minstack。

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stack = []
        self.minstack = []
        
    def push(self, node):
        self.stack.append(node)
        if not self.minstack or node <= self.minstack[-1]:
            self.minstack.append(node)
    def pop(self):
        if self.stack[-1] == self.minstack[-1]:
            del self.minstack[-1]
        return self.stack.pop()
    def top(self):
        if self.stack:
            return self.stack[-1]
        else:
            return None
    def min(self):
        if self.minstack:
            return self.minstack[-1]
        else:
            return None

方法二：
和方法一类似。

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stack = []
        self.minstack = []
    def push(self, node):
        self.stack.append(node)
        if not self.minstack or node < self.minstack[-1]:
            self.minstack.append(node)            
        else:
            self.minstack.append(self.minstack[-1])
    def pop(self):
        if self.stack:
            self.stack.pop()
            self.minstack.pop()
    def top(self):
        if self.stack:
            return self.stack[-1]
        else:
            return None
    def min(self):
        if self.minstack:
            return self.minstack[-1]
        else:
            return None