珍寶鴨的力扣練習(14):棧操作合集
阿新 • • 發佈:2020-12-20
通用方法:輔助棧
考慮借用一個輔助棧 stack ,模擬 壓入 / 彈出操作的排列。根據是否模擬成功,即可得到結果。
題目合集:
題目1:棧的壓入、彈出序列
輸入兩個整數序列,第一個序列表示棧的壓入順序,請判斷第二個序列是否為該棧的彈出順序。假設壓入棧的所有數字均不相等。例如,序列 {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
題解:構造輔助棧
入棧操作: 按照壓棧序列的順序執行。
出棧操作: 每次入棧後,迴圈判斷 “棧頂元素 == 彈出序列的當前元素” 是否成立,將符合彈出序列順序的棧頂元素全部彈出。
class Solution(object):
def validateStackSequences(self, pushed, popped) :
"""
:type pushed: List[int]
:type popped: List[int]
:rtype: bool
"""
if pushed==[]: return True
if popped==[]: return False
i,j=0,0
stack=[]
while i<=len(pushed)-1:
stack.append( pushed[i])
while stack[-1]==popped[j]:
stack.pop()
j+=1
if j==len(popped): return True
if not stack: break
i+=1
return False