1. 程式人生 > >劍指Offer-31棧的壓入、彈出序列

劍指Offer-31棧的壓入、彈出序列

題目:

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

解答:

# -*- coding:utf-8 -*-
class Solution:
    def IsPopOrder(self, pushV, popV):
        # write code here
        stack = []
        i,
j = 0, 0 while(i<len(pushV)): if pushV[i] != popV[j]: stack.append(pushV[i]) i += 1 else: i += 1 j += 1 while(j < len(popV)): if stack.pop() != popV[j]: return False
else: j += 1 return True