题目描述:
输入两个整数序列,第一个序列来表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如,序列 {1,2,3,4,5} 是某栈的压栈序列,序列 {4,5,3,2,1} 是该压栈序列对应的一个弹出序列,但 {4,3,5,1,2} 就不可能是该压栈序列的弹出序列。
思路:这一道题我们可以找一个辅助栈来帮助我们进行压入和弹出的操作,大致思路就是依次把每个元素都先压入栈,也可以随时出栈,但是出栈是有条件的,只有当前栈里面有元素而且栈顶元素就等于出栈队列的对应的元素,那才可以出栈。就是我们会循环着将压入队列放入栈,放入的循环中,我们可以选择出栈也可以不出,如果弹出序列的第一个元素等于了当时的栈顶元素,那表示可以出栈,最后判断是否是弹出序列的条件是判断当前辅助栈是否为空。
代码:
class Solution:
def validateStackSequences(self, pushed: List[int], popped: List[int]) -> bool:
#引入一个辅助栈,元素随时可以出入栈,但是在只有该元素写入栈中并且正好处于栈顶的时候,才可以弹出,最后只需要判断栈是否全部弹出,没有全部弹出说明弹出队列是有问题的
stack=[]
i=0#用来访问弹出序列的元素
for a in pushed:
stack.append(a)#将元素放入栈中
#出栈是有一定的要求的,只有栈顶元素正好等于弹出序列的时候,才会允许出栈,这样才符合栈的原理
while stack and stack[-1]==popped[i]:
stack.pop()#满足条件才会出栈
#接着判断下一个栈顶元素是否可以出栈
i+=1
return not stack#栈空才是正确的,所以取反