題目描述:
輸入兩個整數序列,第一個序列表示棧的壓入順序,請判斷第二個序列是否爲該棧的彈出順序。
假設壓入棧的所有數字均不相等。例如,序列 {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
示例 2:
輸入:pushed = [1,2,3,4,5], popped = [4,3,5,1,2]
輸出:false
解釋:1 不能在 2 之前彈出。
class Solution:
def validateStackSequences(self, pushed: List[int], popped: List[int]) -> bool:
# 做一個輔助棧stack,將pushed中的元素x挨個送入stack
# 如果送入的元素x 就是popped棧頂元素,那麼再將當前元素從stack中取出,
同時popped索引向下走一位
stack=[]
j=0
for x in pushed:
stack.append(x)
while stack and j < len(popped) and stack[-1]==popped[j]:
stack.pop()
j+=1
return not stack