题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
思路(转自 牛客网Alias):
借用一个辅助的栈,遍历压栈顺序,先讲第一个放入栈中,这里是1,然后判断栈顶元素是不是出栈顺序的第一个元素,这里是4,很显然1≠4,所以我们继续压栈,直到相等以后开始出栈,出栈一个元素,则将出栈顺序向后移动一位,直到不相等,这样循环等压栈顺序遍历完成,如果辅助栈还不为空,说明弹出序列不是该栈的弹出顺序。
举例:
入栈1,2,3,4,5
出栈4,5,3,2,1
首先1入辅助栈,此时栈顶1≠4,继续入栈2
此时栈顶2≠4,继续入栈3
此时栈顶3≠4,继续入栈4
此时栈顶4=4,出栈4,弹出序列向后一位,此时为5,,辅助栈里面是1,2,3
此时栈顶3≠5,继续入栈5
此时栈顶5=5,出栈5,弹出序列向后一位,此时为3,,辅助栈里面是1,2,3

自己写的代码,可能比较乱,需要注意python中list.pop([index=-1])用于移除列表中的一个元素(默认最后一个元素),并且返回该元素的值。

def IsPopOrder(self, pushV, popV):
    # write code here
    if len(pushV) != len(popV):
        return False
    stack = []
    while pushV:
        a = pushV.pop(0)
        if a != popV[0]:
            stack.append(a)
        else:
            popV.pop(0)
            if not popV:
                break

            while stack[len(stack)-1] == popV[0]:
                popV.pop(0)
                stack.pop()
                if not stack:
                    break
    if stack:
        return False
    else:
        return True

大神优化版(华科平凡):

def IsPopOrder(self, pushV, popV):
    # write code here
    stack=[]
    while popV:
        ##如果stack的最后一个元素与popV中第一个元素相等,将两个元素都弹出
        if 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

自己优化版:

def IsPopOrder(self, pushV, popV):
    # write code here
    if len(pushV) != len(popV):
        return False
    stack = []
    j = 0
    for i in range(len(pushV)):
        stack.append(pushV[i])
        while j < len(popV) and stack[len(stack)-1] == popV[j]:
            stack.pop()
            j += 1
    if stack:
        return False
    else:
        return True