面试题之检查栈的压入弹出序列(Python实现)
题目描述:输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
思路:利用一个辅助栈模拟栈的压入操作,每次压入时检验是否弹出
python代码实现与相关注释如下:
# coding:utf-8
import sys
def solve():
# 接受压入序列和弹出序列
pushStack = list(map(int, sys.stdin.readline().split()))
popStack = list(map(int, sys.stdin.readline().split()))
# 如果栈为空
if not pushStack or not popStack:
return False
# 辅助栈
stack = []
for i in pushStack:
# 按照压入序列入栈
stack.append(i)
# 每次压入检测是否弹出
while stack and stack[-1] == popStack[0]:
stack.pop()
popStack.pop(0)
# 全部弹出则为正确序列,否则错误
if stack:
return False
else:
return True
if __name__ == '__main__':
# 测试代码
while 1:
print(solve())
如有错误,欢迎交流与指正~