【劍指offer】21—棧的壓入、彈出序列

題目描述

輸入兩個整數序列,第一個序列表示棧的壓入順序,請判斷第二個序列是否可能爲該棧的彈出順序。假設壓入棧的所有數字均不相等。例如序列1,2,3,4,5是某棧的壓入順序,序列4,5,3,2,1是該壓棧序列對應的一個彈出序列,但4,3,5,1,2就不可能是該壓棧序列的彈出序列。(注意:這兩個序列的長度是相等的)

思路:建立一個輔助棧,然後依次將入棧元素壓入,如果棧頂元素等於出棧序列,則循環判斷,最後根據是否爲空,來確定彈出序列正確與否

class Solution:
    def IsPopOrder(self, pushV, popV):
        # write code here    
        stack = []  #先進後出
        j = 0
        for i in range(len(pushV)):
            stack.append(pushV[i])
            # 如果此時的棧頂元素與彈出元素相等,則循環判斷
            while stack and stack[-1]== popV[j]:
                stack.pop()
                j += 1
        # 判斷此時的棧是否爲空
        if stack == []:
            return True
        else:
            return False
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章