剑指offer 之包含min函数的栈

最新推荐文章于 2022-03-18 21:51:16 发布

原创最新推荐文章于 2022-03-18 21:51:16 发布 · 296 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#剑指offer #python #栈

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

17 篇文章

订阅专栏

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

# -*- 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]