包含min函数的栈

题目描述

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

解题思路

要最少的时间复杂度，则用空间换时间。开辟一个新的空间用来存放每次栈中元素的最小值。具体如图所示。

代码示例

# -*- coding:utf-8 -*-
class Solution:
    # 定义栈数据结构的对象
    def __init__(self):
        self.stack = []
        self.minValue = []  # 开辟新的空间用来存储每次栈中最小值

    # 入栈
    def push(self, node):
        # write code here
        self.stack.append(node)
        # 入栈后判断当前栈与当前minValue中栈顶元素的大小，
        # 从而确定是否将当前入栈元素存入minValue中，
        # 如果minValue中有值且入栈的元素比minValue中栈顶元素小，
        # 则说明该元素为此时栈中的最小值，将其存入minValue中
        if self.minValue:
            while self.minValue[-1] > node:
                self.minValue.append(node)
        else:  # minValue为空的情况，直接存入该元素
            self.minValue.append(node)

    # 出栈
    def pop(self):
        # write code here
        # 考虑栈为空的情况，返回None
        if self.stack == []:
            return None
        # stack中直接出栈
        result = self.stack.pop()
        # minValue中如果其栈顶元素与stack出栈的元素相等才出栈
        if self.minValue[-1] == result:
            self.minValue.pop()
        return result

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

    def min(self):
        # write code here
        if self.minValue == []:
            return None
        return self.minValue[-1]

if __name__ == '__main__':
    s = Solution()
    print('----入栈......----')
    s.push(7)
    s.push(8)
    s.push(3)
    s.push(2)
    s.push(9)
    s.push(1)
    print('最小值：',s.min())
    print('----出栈----')
    print(s.pop())
    print(s.pop())
    print(s.pop())
    print(s.pop())
    print(s.pop())
    print(s.pop())