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

输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列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 = []
        j = 0
        for i in pushV:
            stack.append(i)
            while stack:
                if stack and stack[-1] == popV[j]:
                    stack.pop()
                    j += 1
                else:
                    break
        if stack:
            return False
        else:
            return True

思路和我们人脑计算的思路是一致的

  • 建立一个辅助栈 stack = [];
  • 将压入序列的第一个数压入辅助栈,判断辅助栈顶的元素是否和弹出序列的第一个数(当前数)相同,如果相同的话辅助栈弹出该相同数(说明可以弹出序列的第一个数是对的),然后弹出序列移到第二个数(当前数);
  • 如果不同继续将压入序列的数入辅助栈,再次比较辅助栈顶元素是否和弹出序列当前数相同。
  • 直到压入序列中的数全部入辅助栈。
  • 如果辅助栈stack中还有元素,说明不是弹出序列,如果没有说明弹出序列正确。
posted @ 2018-05-21 15:31  Laumians  阅读(386)  评论(1编辑  收藏  举报