题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。
示例:
序列{1,2,3,4,5}是某栈的压栈序列,序列{4,5,3,2,1}是该压栈序列对应的一个弹出序列,
但{4,3,5,1,2}就不可能是该压栈序列的弹出序列。
题目解法
借助一个辅助栈,模拟入栈和出栈的过程:
- 如果下一个弹出的数字恰好是栈顶的数字,那么直接弹出;
- 如果不是,就把压栈序列中还没有入栈的数字压入辅助栈。
结合例子进行说明:
入栈序列[1,2,3,4,5]
出栈序列[4,5,3,2,1]
第一步,1和4不相同,将1压入辅助栈,此时辅助栈为[1];
第二步,2和4不相同,将2压入辅助栈,此时辅助栈为[1,2];
第三步,3和4不相同,将3压入辅助栈,此时辅助栈为[1,2,3];
第四步,4和4相同,将4弹出,此时辅助栈为[1,2,3];
第五步,5和5相同,将5弹出,此时辅助栈为[1,2,3];
第六步辅助栈为[1,2,3],剩下出栈序列为[3,2,1] ,按顺序依次弹出即可。
class Solution:
def IsPopOrder(self, pushV, popV):
# write code here
stack = []
for x in pushV:
stack.append(x)
while stack and stack[-1] == popV[0]:
stack.pop(-1)
popV.pop(0)
return True if len(popV) == 0 else False