【剑指offer】面试题 30：包含 min 函数的栈

最新推荐文章于 2025-11-07 06:50:29 发布

原创最新推荐文章于 2025-11-07 06:50:29 发布 · 394 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

剑指offer

本文介绍了一种利用辅助栈来实现在栈数据结构中快速获取最小元素的方法。通过同步更新数据栈和辅助栈，确保辅助栈顶始终为当前最小值。

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈最小元素的min函数。

时间限制：1秒空间限制：32768K 热度指数：75707

本题知识点：栈

思路

借助辅助栈来保存最小的元素。每次都把最小元素压入辅助栈，那么就能保证辅助栈的栈顶一直都是最小元素。

当最小元素从数据栈内被弹出之后，同时弹出辅助栈的栈顶元素，此时辅助栈的新栈顶元素就是下一个最小值。

Note：有元素要压入数据栈的时候，要判别一下这个元素和辅助栈的栈顶元素的大小，

如果该元素比辅助栈的栈顶元素大，那么该元素压入数据栈，辅助栈的栈顶元素作为最小值再次压入辅助栈；

如果该元素比辅助栈的栈顶元素小，那么该元素压入数据栈，并且将该元素作为最小值压入辅助栈；

参考代码

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stack = []    # 数据栈
        self.minStack = [] # 辅助栈
    def push(self, node):
        # write code here
        self.stack.append(node)
        if self.minStack == [] or node < self.min():
            self.minStack.append(node)
        else:
            temp = self.min()
            self.minStack.append(temp)
            
    def pop(self):
        # write code here
        if self.stack == [] or self.minStack == []:
            return none
        self.stack.pop()
        self.minStack.pop()
        
    def top(self):
        # write code here
        return self.stack[-1]
    
    def min(self):
        # write code here
        return self.minStack[-1]