剑指offer13 --- 包含min函数的栈

最新推荐文章于 2022-04-24 19:01:36 发布

原创最新推荐文章于 2022-04-24 19:01:36 发布 · 114 阅读

0 ·

CC 4.0 BY-SA版权

算法与数据结构专栏收录该内容

38 篇文章

订阅专栏

本文提供了一种实现栈数据结构的方法，其中包含了一个能在O(1)时间复杂度内找到栈中最小元素的min函数。通过使用额外的空间（minValue列表），在每次push操作时更新最小值，确保了min函数的高效性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

剑指offer13 — 包含min函数的栈

题目

定义栈的数据结构，请再该类型中实现一个能够得到栈中所含最小元素的min函数，且满足时间复杂度为O(1)

分析

用时间换空间，或者用空间换时间。为实现时间复杂度O(1),考虑用空间换时间，在该题中即新建一个minValue列表，stack中每pop进一个数值则和minValue[-1]进行一次比较即可知道stack中的最小值，即实现时间复杂度为O(1)的min函数。具体见代码。

解

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() #stack弹出，minValue也应该弹出
        return self.stack.pop()
        
    def min(self):
        if self.minValue == []:
            return None
        return self.minValue[-1]