劍指offer：棧的壓入、彈出序列（Python）

題目描述

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

解題思路

新建一個棧，和一個指向出棧序列第一個元素的指針。按照入棧順序存入序列。當任一時刻棧頂元素和指針指向的出棧序列的第一個元素相同時，彈出該棧頂元素，並右移指針；繼續判斷新的棧頂元素與指針指向的元素是否相同。當不同時繼續按入棧序列將元素入棧，並重覆上述步驟。
這道題自己走了點彎路，想了有點久的時間。還好最後的代碼看起來比較直觀、短小…

Python初稿

def IsPopOrder(self, pushV, popV):
    stack = []
    index = 0
    while pushV:
        stack.append(pushV.pop(0))
        while stack:
            if stack[-1] == popV[index]:
                stack.pop(-1)
                index += 1
            else:
                break
    if not stack:
        return True
    else:
        return False

判斷語句有點繁雜，優化後的代碼如下：

優化後Python

class Solution:
    def IsPopOrder(self, pushV, popV):
        stack = []
        index = 0
        while pushV:
            stack.append(pushV.pop(0))
            while stack and stack[-1] == popV[index]:
                stack.pop(-1)
                index += 1
        return not stack