题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。
例如,序列 {1,2,3,4,5} 是某栈的压栈序列,序列 {4,5,3,2,1} 是该压栈序列对应的一个弹出序列,但 {4,3,5,1,2} 就不可能是该压栈序列的弹出序列。
【解题思路】
如下图所示,给定一个 压入序列 pushed 和 弹出序列 popped ,则压入 / 弹出操作的顺序(即排列)是 唯一确定 的。
如下图所示,栈的数据操作具有 先入后出 的特性,因此某些弹出序列是无法实现的。
考虑借用一个辅助栈 add_stack ,模拟 压入 / 弹出操作的排列。根据是否模拟成功,即可得到结果。
辅助栈 add_stack的出栈入栈如下:
- 入栈操作: 按照压栈序列的顺序执行。
- 出栈操作: 每次入栈后,循环判断 “栈顶元素 == 弹出序列的当前元素” 是否成立,将符合弹出序列顺序的栈顶元素全部弹出。
【算法流程】
- 初始化: 辅助栈 add_stack ,弹出序列的索引 i ;
- 遍历压栈序列pushed: 各元素记为 num ;
- 元素 num 入辅助栈 add_stack;
- add_stack循环出栈:若 stack的栈顶元素 == 弹出序列元素 popped[i] ,则执行出栈与 i++ ;
- 返回值: 若 stack 为空,则此弹出序列合法。
- 复杂度分析:
- 时间复杂度 O(N) : 其中 N 为列表 pushed 的长度;每个元素最多入栈与出栈一次,即最多共 2N 次出入栈操作。
- 空间复杂度 O(N) : 辅助栈 stack 最多同时存储 N 个元素。
class Solution(object):
def validateStackSequences(self, pushed, popped):
add_stack ,i = [],0
for num in pushed:
add_stack.append(num) # num入辅助栈
while add_stack and add_stack[-1]==popped[i]:# 循环判断
add_stack.pop() # 出栈
i+=1
return not add_stack
'''
作者:jyd
链接:https://leetcode-cn.com/problems/zhan-de-ya-ru-dan-chu-xu-lie-lcof/solution/mian-shi-ti-31-zhan-de-ya-ru-dan-chu-xu-lie-mo-n-2/
'''
注:
- 题目指出 pushed 是 popped 的排列 。因此,无需考虑 pushed 和 popped 长度不同 或 包含元素不同 的情况。
- 循环判断辅助栈add_stack是否满足出栈条件的同时,要同时判断add_stack中是否有值