刷题--栈的压入、弹出序列

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

            for j in range(len(popV)):
                while stack[-1] != popV[j]:
                    if i < len(pushV):
                        stack.append(pushV[i])
                        i += 1
                    else:
                        return False

                stack.pop()

        return True

猜你喜欢

转载自blog.csdn.net/treasure_z/article/details/80175199