【Python学习-栈】【剑指offer】之包含min函数的栈

题目

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。
注意：保证测试中不会当栈为空的时候，对栈调用pop()或者min()或者top()方法。

分析

由于时间复杂度是有要求的，所以准备一个新的辅助栈用来保存每次压栈时的最小值，压栈的时候将最小值保存到辅助栈，出栈时同步辅助栈也进行出栈，min时，直接调用辅助栈栈顶元素。
压入4：辅助栈[4]
压入5：辅助栈[4，4]
压入3：辅助栈[4,4,3]
pop: 辅助栈[4,4]
min: 4

代码

class Solution:
    def __init__(self):
        # 数据栈
        self.data = []
        # 辅助栈
        self.helper = []
 
    def push(self, node):
        # write code here
        self.data.append(node)
        #保存最小元素，考虑当前辅助栈为空
        if len(self.helper) == 0 or node <= self.helper[-1]:
            self.helper.append(node) #增加一个元素时，需要与最小值进行比较，如果小于等于最小值，则两个栈都增加该元素，若大于，则数据栈增加该元素，辅助栈增加之前的最小值（前一个栈顶元素）
        #将当前辅助栈中最小元素压入，保持辅助栈和数据栈中元素个数相同
        else:
            self.helper.append(self.helper[-1])
    def pop(self):
        # write code here
        if self.data:
            self.helper.pop()  # 出栈时，两个栈都需要进行删除
            return self.data.pop()
    def top(self):
        # write code here
         if self.data: # 查看栈顶元素 没有涉及删改操作，所以无影响
            return self.data[-1]
 
    def min(self):
        # write code here  # 求最小值时 ，直接输出辅助栈的栈顶元素
        if self.helper:
            return self.helper[-1]

栈的压入、弹出序列

输入两个整数序列，第一个序列表示栈的压入顺序，请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序，序列4,5,3,2,1是该压栈序列对应的一个弹出序列，但4,3,5,1,2就不可能是该压栈序列的弹出序列。（注意：这两个序列的长度是相等的）

分析

看了好几遍，后面看题解才弄懂这个顺序，先解释一下这个顺序：
借助一个辅助栈，将1,2,3,4,5依次压入：
1压入，比较4,5,3,2,1中的第一位4是否相等，不相等，则继续压入
2压入，同上不相等，继续压入
3压入，同上不相等，继续压入
4压入，这时与4,5,3,2,1中的第一位4相等了，那么这个时候4就要出栈了，刚刚压入的4要出栈，4,5,3,2,1中的4也要出栈，也就是只要相等，相等的那个数都要从两个栈同时出栈，出栈后5,3,2,1中的值为5，刚刚压入的4出栈后栈顶元素为3,3不等于5，所以继续压入，此时要压入5
5压入，又与5,3,2,1中的第一个值5相等，则两个5都要出栈，这是栈顶元素为3，与3,2,1中的第一个值3又相等，两者同时出栈，接下来是2，接下来是1，最后都弹出序列了。
下面看一下4,3,5,1,2为什么不行？同样按照刚刚的顺序来走一遍，
1压入，比较4,3,5,1,2中的第一位4是否相等，不相等，则继续压入
2压入，同上不相等，继续压入
3压入，同上不相等，继续压入
4压入，两个4相等，同时出栈，栈顶为3，3,5,1,2中的第一位3再次相等，又一次同时出栈，栈顶为2，与5,1,2中的第一位5不相等，则开始压栈
5压入，与5,1,2中的第一位5相等，两个5同时出栈，此时栈顶元素为2，与1,2中的第一位1不相等，这样就都不能出栈了。

代码

# -*- coding:utf-8 -*-
class Solution:
    def IsPopOrder(self, pushV, popV):
        # write code here
        if not pushV or  len(pushV)!=len(popV): # 说了长度相同，空也直接返回False
            return False
        stack=[] #准备空列表进行压栈
        for i in pushV:
            stack.append(i)  # 一次压栈
            while len(stack) and stack[-1]==popV[0]:  #栈中有元素且栈顶元素与第一个元素相等，相等元素从两个栈中出栈，
                stack.pop()
                popV.pop(0)
       
        if len(stack): # 遍历完后，栈中还有元素存在，那么这种顺序也是不符合的，这里就是对第二种情况的判断
            return False
        return True