Leetcode刷题：剑指offer【面试题31 栈的压入、弹出序列】

【面试题31 栈的压入、弹出序列】

难度： 中等

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

Leetcode题目对应位置： 面试题31：栈的压入、弹出序列

思路：遍历

按照 pushed 列表元素顺序进行入栈，每次入栈后判断当前栈顶元素是否为要出栈元素，若是则出栈，继续循环判断当前栈顶元素是否为要出栈元素，如果需要出栈的元素不在栈顶，则把压栈序列中还未入栈的数字入栈，直到把下一个需要弹出的数字压入栈顶为止。如果所有数字都已入栈后仍未找到下一个需要出栈的数字，则该出栈序列不可能存在。

特殊输入：当 pushed 和 popped 长度不等时，直接返回 False；当 pushed 和 popped 元素完全一致时，直接返回 True。
判断条件：当入栈所有元素并出栈后，若辅助栈为空，则说明顺序正确。

时间复杂度： O(n)，其中 n 为列表 pushed 的长度；每个元素最多入栈与出栈一次，即最多共 2n次出入栈操作。
空间复杂度： O(n)，辅助栈 stack 最多同时存储 n 个元素。

class Solution:
    def validateStackSequences(self, pushed, popped):
        p1 = len(pushed)
        p2 = len(popped)
        if p1 != p2: return False
        if pushed == popped: return True
        stack = []
        j = 0
        for i in range(p1):
            stack.append(pushed[i])
            while j < p2 and stack:
                if stack[-1] == popped[j]:
                    stack.pop()
                    j += 1
                else: break
        return True if not stack else False

看到别人写的简洁代码，学习学习：

class Solution:
    def validateStackSequences(self, pushed, popped):
        stack, i = [], 0
        for n in pushed:
            stack.append(n)
            while stack and stack[-1] == popped[i]:
                    stack.pop()
                    i += 1
        return not stack

代码来源：solution/mian-shi-ti-31-zhan-de-ya-ru-dan-chu-xu-lie-mo-n-2