题目:
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如,序列 {1,2,3,4,5} 是某栈的压栈序列,序列 {4,5,3,2,1} 是该压栈序列对应的一个弹出序列,但 {4,3,5,1,2} 就不可能是该压栈序列的弹出序列。
LeetCode原题:栈的压入、弹出序列
思路:
第一个序列表示栈的压入顺序,因为在压入过程中,随时可能从当前序列弹出,所以可以构建一个辅助栈,模拟栈的压入过程,设置两个指针,分别指向第一个压入序列和第二个弹出序列,只要当前的辅助栈的栈顶元素不等于弹出序列指针的对应的元素,辅助栈一直从压入序列压入新的数。当两者相等时,说明此时需要弹出,从辅助栈弹出,同时弹出序列指针+1。
当所有的数都已经(有可能被中途弹出)被压入到辅助栈中,而此时弹出序列的指针没有到尾部,说明一直没有数能和弹出序列指针对应的数匹配成功,返回False;否则都匹配成功,返回True。
代码:
class Solution:
def validateStackSequences(self, pushed: List[int], popped: List[int]) -> bool:
if len(pushed) == 0 and len(popped) == 0: return True
if len(pushed) == 0 or len(popped) == 0: return False
stack = []
stack.append(pushed[0])
index = 1
p_index = 0
while p_index < len(popped):
while len(stack) == 0 or (index < len(pushed) and stack[-1] != popped[p_index]):
stack.append(pushed[index])
index += 1
if stack[-1] != popped[p_index]:
break
stack.pop()
p_index += 1
if len(stack) == 0 and p_index == len(popped):
return True
else:
return False