判断第二个序列是否为第一个序列的弹出

题目:

输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如,序列 {1,2,3,4,5} 是某栈的压栈序列,序列 {4,5,3,2,1} 是该压栈序列对应的一个弹出序列,但 {4,3,5,1,2} 就不可能是该压栈序列的弹出序列。
LeetCode原题:栈的压入、弹出序列

思路:
第一个序列表示栈的压入顺序,因为在压入过程中,随时可能从当前序列弹出,所以可以构建一个辅助栈,模拟栈的压入过程,设置两个指针,分别指向第一个压入序列和第二个弹出序列,只要当前的辅助栈的栈顶元素不等于弹出序列指针的对应的元素,辅助栈一直从压入序列压入新的数。当两者相等时,说明此时需要弹出,从辅助栈弹出,同时弹出序列指针+1。
当所有的数都已经(有可能被中途弹出)被压入到辅助栈中,而此时弹出序列的指针没有到尾部,说明一直没有数能和弹出序列指针对应的数匹配成功,返回False;否则都匹配成功,返回True。

代码:

class Solution:
    def validateStackSequences(self, pushed: List[int], popped: List[int]) -> bool:
        if len(pushed) == 0 and len(popped) == 0: return True
        if len(pushed) == 0 or len(popped) == 0: return False

        stack = []
        stack.append(pushed[0])
        index = 1
        p_index = 0

        while p_index < len(popped):
            while len(stack) == 0 or (index < len(pushed) and stack[-1] != popped[p_index]):
                stack.append(pushed[index])
                index += 1

            if stack[-1] != popped[p_index]:
                break

            stack.pop()
            p_index += 1

        if len(stack) == 0 and p_index == len(popped):
            return True
        else:
            return False

猜你喜欢

转载自blog.csdn.net/yaogepila/article/details/107887077