题目:
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如,序列 {1,2,3,4,5} 是某栈的压栈序列,序列 {4,5,3,2,1} 是该压栈序列对应的一个弹出序列,但 {4,3,5,1,2} 就不可能是该压栈序列的弹出序列。
leetcode原题链接
思路:
参考Krahets题解
给定一个压入序列 pushedpushed 和弹出序列 poppedpopped ,则压入 / 弹出操作的顺序(即排列)是 唯一确定 的。
- 借用辅助栈stack,模拟栈的压入、弹出,模拟成功return True,模拟失败return False
- stack的入栈操作按pushed列表顺序执行,入栈过程中,如果栈顶元素(即stack[-1])等于poped出栈列表的当前元素,则证明,该元素此时应执行弹出操作。将该元素从辅助栈和poped列表中删除,继续对下一个元素进行操作(入栈并判断)
- 模拟成功:pushed中全部元素均入辅助栈stack,并且按poped列表的顺序 弹出,最终stack应为空
代码:
class Solution:
def validateStackSequences(self, pushed: List[int], popped: List[int]) -> bool:
stack=[]#辅助栈
while pushed:
stack.append(pushed.pop(0))
while stack and stack[-1]==popped[0]:
stack.pop()
popped.pop(0)
if len(stack):#stack中仍有值
return False
return True#stack==[]
- 优化代码
class Solution:
def validateStackSequences(self, pushed: List[int], popped: List[int]) -> bool:
stack=[]
for i in pushed:#用for循环按序遍历 代替while循环+pop操作
stack.append(i)
while stack and stack[-1]==popped[0]:
stack.pop()
popped.pop(0)
return not stack