1. 程式人生 > >【劍指Offer】21棧的壓入、彈出序列

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

題目描述

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

時間限制:1秒;空間限制:32768K;本題知識點:棧

解題思路

PUSH 1 2 3 4 5

POP  4 5 3 2 1 正確

         4 3 5 1 2 錯誤

         4 3

5 2 1 正確 (4和43看作一樣的情況)

# -*- coding:utf-8 -*-
class Solution:
    def IsPopOrder(self, pushV, popV):
        # write code here
        if not pushV 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