题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
解法1:提炼判断逻辑
看了解析,大部分人的想法是模拟入栈和出栈的过程,这需要一个辅助栈,会稍微费内存一些
我的想法是通过入栈出栈的过程,提炼出逻辑判断的标志,这样就不需要再额外创建一个辅助栈了
判断的逻辑很简单:
首先将出栈的第一个元素当作pointer(指向pushV对应的位置)
第一个元素肯定能被弹出,所以初始化指针是不会出错的(除非出现元素不对应的情况,try的部分就是针对这种情况的)
初始化完成后开始迭代判断
设定index为将要被弹出元素在pushV中的位置(可以看做元素指针)
通过观察出入栈的过程,得到判断是否能完成出栈必须满足以下逻辑:
如果index < pointer - 1,则弹出不可能
如果这个元素能被弹出,那么:
pointer = index
注意操作这个步骤的时候,两个list的内容的变化
判断逻辑是怎么得出的:
如果一个元素只有在两种情况下能被弹出:
1,比pointer前一个压入
2,比pointer后压入
所以整个代码里最核心的一句就是:
if index < pointer - 1:
return False
源码:
# -*- coding:utf-8 -*-
class Solution:
def IsPopOrder(self, pushV, popV):
# write code here
flag = popV.pop(0)
try:
pointer = pushV.index(flag)
except:
return False
for i in range(len(popV)):
element = popV.pop(0)
try:
index = pushV.index(element)
except:
return False
if index < pointer - 1:
return False
pushV.pop(pointer)
pointer = pushV.index(element)
return True