题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列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