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

来源:互联网 发布:淘宝买电脑可靠吗 编辑:程序博客网 时间:2024/05/18 01:27

题目描述

定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的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]







原创粉丝点击