[剑指offer] 包含min函数的栈

题目内容

https://www.nowcoder.com/practice/4c776177d2c04c2494f2555c9fcc1e49?tpId=13&tqId=11173&tPage=1&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

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

---

题目思路

这道题目需要构建一个最小栈和普通栈。普通栈用来进入弹出元素，而最小栈用来保留每一次的最小元素。

---

程序代码

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stack=[]
        self.min_stack=[]
    def push(self, node):
        # write code here
        self.stack.append(node)
        if not self.min_stack:
            self.min_stack.append(node)
        else:
            self.min_stack.append(min(node,self.min_stack[-1]))#在当前最小元素和插入元素中选择最小的那个
        
    def pop(self):
        # write code here
        self.min_stack.pop()
        return self.stack.pop()
    def top(self):
        # write code here
        return self.stack[-1]
    def min(self):
        # write code here
        return self.min_stack[-1]