剑指offer 之 栈的压入、弹出序列是否对应

# -*- coding:utf-8 -*-
'''
面试22：输入的两个整数序列、第一个表示栈的压入顺序，判断第二个序列是不是该栈的弹出序列
基本思路：
    (1).建立一个辅助栈，把push序列的数字依次压入辅助栈，
    (2).每次压入后，比较辅助栈的栈顶元素和pop序列的首元素是否相等，
	    相等的话， 就继续比较pop序列的下一元素和辅助栈的栈顶元素，
	(3).若最后辅助栈为空，则push序列可以对应于pop序列。
'''
class Solution:
    def IsPopOrder(self, pushV, popV):
        # write code here
        if not pushV or not popV:
        	return False
        as_stack = []
        n = len(pushV)
        j = 0

        for i in pushV:
        	# 压入栈
        	as_stack.append(i)
        	while len(as_stack) and as_stack[-1] == popV[j]:	
        		as_stack.pop()
        		j += 1

        if as_stack:
        	return False
        return True

# 测试用例
pushV = [1, 2, 3, 4, 5]
popV = [4, 5, 3, 2, 1]
popVF = [4, 5, 2, 1, 3]
S = Solution()
print(S.IsPopOrder(pushV, popV))
print(S.IsPopOrder(pushV, popVF))