描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列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):
j = 0
stack = [] #初始化一个辅助栈stack
for x in pushV: #把压入序列中的元素依次添加到辅助栈中
stack.append(x)
while stack and stack[-1] == popV[j]: #如果辅助栈的 栈顶元素就等于弹出栈的j 位对应元素
stack.pop() #弹出辅助栈该元素
j += 1 #j 累加
return j == len(popV) #最后直到j等于弹出序列长度结束该循环
大体是建立一个辅助栈,将数组压入然后通过比较,如果辅助栈的顶部,也就是第一序列的输出第一个元素跟第二个序列的首元素相同,就继续循环,同时将辅助栈的顶部元素pop,第二序列的索引加1。