题目来源:https://leetcode-cn.com/problems/zhan-de-ya-ru-dan-chu-xu-lie-lcof/
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如,序列 {1,2,3,4,5} 是某栈的压栈序列,序列 {4,5,3,2,1} 是该压栈序列对应的一个弹出序列,但 {4,3,5,1,2} 就不可能是该压栈序列的弹出序列。
示例 1:
输入:pushed = [1,2,3,4,5], popped = [4,5,3,2,1]
输出:true
解释:我们可以按以下顺序执行:
push(1), push(2), push(3), push(4), pop() -> 4,
push(5), pop() -> 5, pop() -> 3, pop() -> 2, pop() -> 1
思路:
以前微机课讲过的题,思路是一样的,将pushed的元素按顺序入栈,同时对比popped的元素,如果栈底元素等于popped元素,则该元素出栈。如果popped是一种出栈顺序,那么最后栈肯定是空的。否则就不是。
代码:
class Solution(object):
def validateStackSequences(self, pushed, popped):
"""
:type pushed: List[int]
:type popped: List[int]
:rtype: bool
"""
stack = []
j = 0
for x in pushed:
stack.append(x)
while stack and j < len(popped) and popped[j] == stack[-1]:
stack.pop()
j += 1
return not stack