20180722剑指offer题21——包含min函数的栈

一、要求

    栈的push，pop，min操作的时间复杂度都是O(1)

二、思路及代码

    用两个list实现，stack用来做正常的入栈出栈，minStack用来保存当前栈的最小值。需要注意的是，minStack不是对所有栈中的数据进行排序 ：如果push的数字小于minStack栈顶的数字，在minStack中push更小的新数据；否则，在minStack中push把旧的最小数（即原来minStack最后一个元素）再次push一次，这样minStack中记录的就是每增减一个数后的最小值，minStack[i]对应的是stack[0:i+1]的最小值，minStack长度和stack一样大。push和pop是常规操作，不展开了。

class Solution():
    def __init__(self):
        self.stack=[]
        self.minStack=[]
    def push(self,num):
        self.stack.append(num)
        if self.minStack==[] or num<self.minStack[-1]:#当栈为空或新数小于原最小数时，直接压入最小栈
            self.minStack.append(num)
        else:
            self.minStack.append(self.minStack[-1])

    def pop(self):
        if self.stack:
            self.stack.pop()
            self.minStack.pop()

    def min(self):
        if self.minStack:
            return self.minStack[-1]
        else:
            return None




S = Solution()
S.push(3)
S.push(4)
S.push(2)
S.push(1)
print(S.min())
S.pop()
print(S.min())
S.pop()
print(S.min())

三、运行结果

1
2
3

四、思考与总结

无