剑指Offer：包含min函数的栈（Python）

实现含min函数的栈

最新推荐文章于 2020-03-20 18:56:04 发布

原创最新推荐文章于 2020-03-20 18:56:04 发布 · 252 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#栈 #python #剑指offer

剑指Offer 专栏收录该内容

21 篇文章

订阅专栏

本文介绍了一种特殊栈的数据结构实现，通过维护两个栈，其中一个用于存储数据，另一个用于实时跟踪并存储当前栈中的最小值，实现了在O(1)时间复杂度下获取栈中最小元素的功能。

题目

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数。时间复杂度应为O（1）

思路

空间换时间
另一个栈一直存储最小值

解答

方法一

class Solution:
    def __init__(self):
        self.stack = []
        self.minValue = []
    def push(self, node):
        self.stack.append(node)
        if self.minValue:
            if self.minValue[-1] > node:
                self.minValue.append(node)
            else:
                self.minValue.append(self.minValue[-1])
        else:
            self.minValue.append(node)
    def pop(self):
        if self.stack == []:
            return None
        self.minValue.pop()
        return self.stack.pop()
    def top(self):
        if self.stack == []:
            return None
        return self.stack[-1]
    def min(self):
        if self.minValue == []:
            return None
        return self.minValue[-1]