面試題31：棧的壓入、彈出序列

題目描述

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

思路：定義一個輔助棧，並把壓棧序列的數字依次壓入輔助棧。 如果下一個彈出的數字剛好是棧頂數字，那麼直接彈出。如果下一個彈出的數字不在棧頂，則把壓棧序列中還沒有入棧的數字壓入輔助棧，直到把下一個需要彈出的數字壓入棧頂爲止。如果所有數字都壓入棧頂後仍然沒有找到下一個彈出的數字，那麼該序列不可能是一個彈出序列。

# 判斷序列是否爲該棧的彈出順序
class Solution:
    def IsPopOrder(self, pushV, popV):
        # 判斷
        if not pushV or not popV or len(pushV)!=len(popV):
            return False
        # 輔助棧
        stack = []
        for i in pushV: # 數據加入輔助棧
            stack.append(i)
            while len(stack) and stack[-1]==popV[0]:
                stack.pop()
                popV.pop(0)
        if len(stack):
            return False
        return True

if __name__=="__main__":
    pushV = [1,2,3,4,5]
    popV = [4,5,3,2,1]
    solution = Solution()
    print(solution.IsPopOrder(pushV, popV))