题目描述:
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
思路:
使用一个辅助栈来模拟出入栈的过程。
首先,假设入栈顺序是1,2,3,4
正确的出栈顺序(其中一种)2,3,4,1
错误的出栈顺序(其中一种)3,1,4,2
然后,我们设置一个中间栈:tempStack
1、按入栈顺序,存入一个元素到tempStack中
2、比较tempStack的栈顶元素与出栈顺序的第index(从出栈顺序第一个开始)个元素比较
相同,则进行3,否则进行1
3、弹出tempStack的栈顶元素,index++。执行2。
举例:入栈顺序1,2,3,4,出栈顺序2,3,4,1
将1存入tempStack,index=0,1不等于2,接着将2存入tempStack,2等于2,说明出入栈过程是把1压进去,放入2后又取出;2出栈,index=1,1不等于3,接着将3存入tempStack,3等于3,说明3入栈后又出栈;3出栈,index=2,4不等于1,接着将4存入tempStack,4等于4,说明4入栈后出栈;4出栈,index=3,1等于1,最后1出栈,tempStack为空,出栈顺序正确。
实现代码:
# -*- coding:utf-8 -*-
class Solution:
def IsPopOrder(self, pushV, popV):
# write code here
i = 0
temp_stack = []
for push in pushV:
temp_stack.append(push)
while len(temp_stack) and popV[i] == temp_stack[-1]:
temp_stack.pop()
i += 1
if not temp_stack:
return True
else:
return False