剑指offer栈的压入、弹出序列python

题目描述

输入两个整数序列，第一个序列表示栈的压入顺序，请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序，序列4,5,3,2,1是该压栈序列对应的一个弹出序列，但4,3,5,1,2就不可能是该压栈序列的弹出序列。（注意：这两个序列的长度是相等的）

 

思路

用一个指针来遍历弹出序列

如果当前所指数字等于栈顶，弹出栈顶数字，弹出序列指针后移

如果不等于的话或者栈为空的时候，则将压入序列入栈

如果压入序列为空了，弹出序列却不为空，而且栈顶数字不等于当前所指数字，则不能。

代码

# -*- coding:utf-8 -*-
class Solution:
    def IsPopOrder(self, pushV, popV):
        # write code here
        stack = []
        i = 0
        l = len(popV)
        while i < l:
            if stack:
                if stack[-1] == popV[i]:
                    stack.pop() 
                    i+= 1
            if pushV:
                stack.append(pushV.pop(0))
            if stack:
                if (not pushV) and stack[-1] != popV[i]:
                    return False
        return True