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