2.8 剑指offer 栈的压入、弹出序列

题目

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

思路

通过题目的例子，画图总结规律，使用一个辅助栈。
按弹出栈的元素压入并弹出输入序列的元素到辅助栈中。

• 如果下一个弹出的数字刚好是辅助栈栈顶数字，那么直接弹出；
• 如果下一个弹出的数字不在栈顶，则把输入序列中还没有入栈的数字压入辅助栈，直到把下一个需要弹出的数字压入栈顶为止；
• 如果所有数字都压入栈后，仍然没有找到下一个弹出的数字，那么该序列不可能是一个弹出序列。

代码

def IsPopOrder(pushV, popV):
    if len(pushV) == 0 or len(popV) == 0 or len(pushV) != len(popV):
        return False
    assist =[]
    pushHead = -1
    while len(pushV) != 0:
        popHead = popV.pop(0) # 最好统一pop，或[0]取值，不容易乱
        pushHead = pushV[0]
        if popHead == pushHead:
            pushV.pop(0)
        else:
            while len(pushV) != 0 and popHead != pushHead:
                temp = pushV.pop(0)
                pushHead = temp
                assist.append(temp)
            if popHead == pushHead: 
            	# 把最后一个压入辅助栈的，和弹出栈顶元素相同的值弹出。
                assist.pop()
            else:
                return False
    for i in range(len(popV)):
        if popV[0] == assist[-1]:
            popV.pop(0)
            assist.pop()
    if len(popV) != 0:
        return False
    return True

以下解法
链接：https://www.nowcoder.com/questionTerminal/d77d11405cc7470d82554cb392585106
来源：牛客网

    def IsPopOrder(self, pushV, popV):
        stack = []
        for psh in pushV:
            stack.append(psh)
            if stack[-1] == popV[0]:
                popV.pop(0)
                stack.pop()
        for pp in popV:
            if pp == stack[-1]:
                stack.pop()
        return stack == []
        
    def IsPopOrder(self, pushV, popV):
        # stack中存入pushV中取出的数据
        stack=[]
        while popV:
            # 如果第一个元素相等，直接都弹出，根本不用压入stack
            if pushV and popV[0]==pushV[0]:
                popV.pop(0)
                pushV.pop(0)
            #如果stack的最后一个元素与popV中第一个元素相等，将两个元素都弹出
            elif stack and stack[-1]==popV[0]:
                stack.pop()
                popV.pop(0)
            # 如果pushV中有数据，压入stack
            elif pushV:
                stack.append(pushV.pop(0))
            # 上面情况都不满足，直接返回false。
            else:
                return False
        return True