题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
思路 1 最开始的思路,寻找规律
通过举例 压入栈为 1,2,3,4,5
弹出栈 3,4,5,2,1; 3,5,4,2,1都为该栈的弹出顺序
而3,5,4,1,2不符合要求
可总结规律为:
- 弹出栈的第一个元素可将压入栈分为左右两个部分,
例如压入栈为 1,2,3,4,5;弹出栈 3,4,5,2,1;则3将压入栈分为了1,2和4,5两个部分。 - 由栈的先进后出特点可知:
#!/usr/bin/env python
# _*_ coding:utf-8 _*_
# -*- coding:utf-8 -*-
class Solution:
def IsPopOrder(self, pushV, popV):
# write code here
if popV[0] not in pushV:
return False
index = pushV.index(popV[0])
if index == 0:
return pushV == popV
else:
for i in pushV[(index+1):]:
if i not in popV[1:(1+len(pushV[(index+1):]))]:
return False
return pushV[(index-1)::-1]== popV[1+len(pushV[(index+1):]):]
s = Solution()
print(s.IsPopOrder([1,2,3,4,5],[1,2,3,4,5]))
思路 2 借鉴了 剑指offer 大神优化版(华科平凡)
举例:压入栈为 1,2,3,4,5;弹出栈 3,4,5,2,1;
利用辅助栈的思维:
将压入栈从底部弹出压入辅助栈,这里是1,在压入过程中,判断元素1是否和弹出栈栈底元素3相等,很明显,这不相等,继续向辅助栈依次添加元素2,3,4,5,如果相等,就弹出辅助栈栈头和弹出栈的栈底,直到弹出栈为空。等压栈顺序遍历完成后,辅助栈为空则说明是该压栈序列的弹出序列。
换种说法就是利用辅助栈拷贝了了一份压入栈,在压入的过程中依次和弹出栈作比较,相同则出栈,不同则压栈,知道压栈遍历完。
class Solution:
def IsPopOrder(self, pushV, popV):
# write code here
assist_stack = []
while popV:
#如果assist_stack的最后一个元素和popV的第一个元素相同,则将assist_stack和popV弹出
if assist_stack and popV[0] == assist_stack[-1]:
assist_stack.pop()
popV.pop(0)
#如果pushV压入assist_stack中
elif pushV:
assist_stack.append(pushV.pop(0))
else:
return False
return True