题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
思路(转自 牛客网Alias):
借用一个辅助的栈,遍历压栈顺序,先讲第一个放入栈中,这里是1,然后判断栈顶元素是不是出栈顺序的第一个元素,这里是4,很显然1≠4,所以我们继续压栈,直到相等以后开始出栈,出栈一个元素,则将出栈顺序向后移动一位,直到不相等,这样循环等压栈顺序遍历完成,如果辅助栈还不为空,说明弹出序列不是该栈的弹出顺序。
举例:
入栈1,2,3,4,5
出栈4,5,3,2,1
首先1入辅助栈,此时栈顶1≠4,继续入栈2
此时栈顶2≠4,继续入栈3
此时栈顶3≠4,继续入栈4
此时栈顶4=4,出栈4,弹出序列向后一位,此时为5,,辅助栈里面是1,2,3
此时栈顶3≠5,继续入栈5
此时栈顶5=5,出栈5,弹出序列向后一位,此时为3,,辅助栈里面是1,2,3
自己写的代码,可能比较乱,需要注意python中list.pop([index=-1])用于移除列表中的一个元素(默认最后一个元素),并且返回该元素的值。
def IsPopOrder(self, pushV, popV):
# write code here
if len(pushV) != len(popV):
return False
stack = []
while pushV:
a = pushV.pop(0)
if a != popV[0]:
stack.append(a)
else:
popV.pop(0)
if not popV:
break
while stack[len(stack)-1] == popV[0]:
popV.pop(0)
stack.pop()
if not stack:
break
if stack:
return False
else:
return True
大神优化版(华科平凡):
def IsPopOrder(self, pushV, popV):
# write code here
stack=[]
while popV:
##如果stack的最后一个元素与popV中第一个元素相等,将两个元素都弹出
if stack and stack[-1]==popV[0]:
stack.pop()
popV.pop(0)
# 如果pushV中有数据,压入stack
elif pushV:
stack.append(pushV.pop(0))
# 上面情况都不满足,直接返回false。
else:
return False
return True
自己优化版:
def IsPopOrder(self, pushV, popV):
# write code here
if len(pushV) != len(popV):
return False
stack = []
j = 0
for i in range(len(pushV)):
stack.append(pushV[i])
while j < len(popV) and stack[len(stack)-1] == popV[j]:
stack.pop()
j += 1
if stack:
return False
else:
return True