题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
Python解答
本题的思路是创建一个辅助栈,先挨个比较原先的序列(栈的压入顺序),再挨个比较辅助栈的序列。
# -*- coding:utf-8 -*-
class Solution:
def __init__(self):
self.stack = [] # 初始化辅助栈
def IsPopOrder(self, pushV, popV):
# write code here
while pushV: # 若栈的压入序列不为空,则将压入序列与弹出序列挨个比较
if pushV[0] != popV[0]: # 若压入序列的开头与弹出序列的开头不相等,则把压入序列的开头元素压到辅助栈中,再比较压入序列第二个元素与弹出序列
self.stack.append(pushV.pop(0))
else: # 若压入序列的开头与弹出序列的开头相等,则同时弹出
pushV.pop(0)
popV.pop(0)
while self.stack: # 若辅助栈的元素不为空,则挨个将栈顶元素与弹出序列剩余的开头元素比较
if self.stack[-1] == popV[0]: # 若相等,则同时pop掉
self.stack.pop() == popV.pop(0)
else: # 若不等,则不为压入序列的弹出序列
return False
return True # 若压入序列与辅助栈的元素均为空,则输出True