剑指offer 之包含min函数的栈

chimuuu

于 2017-07-30 21:10:16 发布

阅读量278

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指offer 之 Python实现文章标签：剑指offer python 栈

本文链接：https://blog.youkuaiyun.com/C_Donald/article/details/76407434

剑指offer 之 Python实现专栏收录该内容

17 篇文章

订阅专栏

本文介绍了一种特殊栈的设计方案，该栈除了具备常规的压栈和弹栈功能外，还提供了一个min函数用于返回栈中当前的最小元素。通过维护两个栈，一个用于存储所有元素，另一个专门用来跟踪最小值，从而实现了高效地获取最小元素。

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

# -*- coding:utf-8 -*-

'''
面试21：包含min函数的栈
题目描述
定义栈的数据结构，请在该类型中实现一个能够得到栈最小元素的min函数。 
'''

class Solution:
    def __init__(self):
        self.stack = []
        self.stack_min = []

    def push(self, node):
        # write code here
        if not self.stack_min:
        	self.stack_min.append(node)
       	else:
       		if self.min() <= node:
       			self.stack_min.append(self.min())
       		else:
       			self.stack_min.append(node)
       	self.stack.append(node)

    def pop(self):
        # write code here
        if not self.stack:
        	return []
        else:
        	self.stack_min.pop()
        	return self.stack.pop()

    def top(self):
        # write code here
        if not self.stack:
        	return []
        return self.stack[-1]

    def min(self):
        # write code here
        if not self.stack_min:
        	return []
        else:
        	return self.stack_min[-1]