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

Solutions

  首先要理解题意,这里用辅助栈的形式存储最小的结果。

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.data = []
        self.mins = []
    def push(self, node):
        # write code here
        self.data.append(node)
        
        if len(self.mins) > 0 and self.mins[-1] < node:
            self.mins.append(self.mins[-1])
        else:
            self.mins.append(node)
    def pop(self):
        # write code here
        assert len(self.data) > 0 and len(self.mins) > 0
        self.data.pop()
        self.mins.pop()
    def top(self):
        # write code here
        return self.data[-1]
    def min(self):
        # write code here
        assert len(self.data) > 0 and len(self.mins) > 0
        return self.mins[-1]

References

  1. 021. 包含 min 函数的栈