面试题之检查栈的压入弹出序列(Python实现)

面试题之检查栈的压入弹出序列(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())

如有错误，欢迎交流与指正~