给定 pushed
和 popped
两个序列,只有当它们可能是在最初空栈上进行的推入 push 和弹出 pop 操作序列的结果时,返回 true
;否则,返回 false
。
示例 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
示例 2:
输入:pushed = [1,2,3,4,5], popped = [4,3,5,1,2]
输出:false
解释:1 不能在 2 之前弹出。
提示:
0 <= pushed.length == popped.length <= 1000
0 <= pushed[i], popped[i] < 1000
pushed
是 popped
的排列。解题思路
这个问题非常简单,模拟题目的入栈出栈操作就可以了。当pushed
不为空的时候,我们弹出pushed
的第一个元素加入到tmp
中
我们发现此时的tmp
最后一个元素和popped
中的第一个元素不一样
我们继续压入pushed
中的元素。
我们发现此时tmp[-1]==popped[0]
,所以我们进行弹出操作,分别弹出两个相同的元素。
我们发现此时的元素又不相同了,我们再次压入pushed
中的第一个元素。
接着就是一系列的弹出操作。我们最后只要判断tmp
中的元素是不是空即可确定栈顺序是不是合理。
class Solution:
def validateStackSequences(self, pushed: List[int], popped: List[int]) -> bool:
tmp = list()
while pushed:
p = pushed.pop(0)
tmp.append(p)
while tmp and popped[0] == tmp[-1]:
popped.pop(0)
tmp.pop()
return not tmp
也可以在O(1)
的空间复杂度内完成,只需通过一个变量记录栈的大小即可。
class Solution:
def validateStackSequences(self, pushed: List[int], popped: List[int]) -> bool:
i = j = 0
for x in pushed:
pushed[i] = x
i += 1
while i > 0 and pushed[i - 1] == popped[j]:
i, j = i - 1, j + 1
return i == 0
此时我们将原数组pushed
作为栈,i++
表示入栈操作,i--
表示出栈操作。
我将该问题的其他语言版本添加到了我的GitHub Leetcode
如有问题,希望大家指出!!!