题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
输入输出示例
输入:[1,2,3,4,5],[4,3,5,1,2]
返回值:false
解题思路
每去压入一个栈的时候,都去判断是否与当前弹出的栈一样,如果一样就弹出,如果不一样就继续压入。
- 首先需要一个栈
- 按照pushV的方式入栈,
- 判断每次入栈后栈顶元素是否与弹出序列中当前指向的元素相等,如果相等则将该元素从栈中弹出,如果不相等则继续进行压栈操作。
- 弹出的时候需要循环判断是否需要弹出
- 判断是否需要弹出的时机:刚刚压入过后就需要判断,
- 判断需要弹出的情况的条件:压入栈的顶部和弹出栈的顶部数据相等
代码示例
# coding:utf-8
'''
栈的压入、弹出序列:
根据给出的压入序列,判断是否满足栈的弹出序列
'''
class Solution:
def isPopOrder(self, pushV, popV):
# 首先定义一个栈
stack =[]
# 考虑返回None的情况:
# 入栈序列为空或者入栈序列与弹出序列长度不一致
if pushV == [] or len(pushV) != len(popV):
return None
# 循环入栈序列进行压栈操作
index = 0
for item in pushV:
stack.append(item)
# 每一次压栈后都需要判断是否需要弹出
# 弹出的条件是:stack中有元素且栈顶元素与弹出序列index所指元素相等
# 要使用while循环的方式,因为每一次弹出后仍要判断后续的元素
while stack and stack[-1] == popV[index]:
stack.pop()
index += 1
# 最终,如果栈为空,说明弹出序列正确,否则证明弹出序列顺序不符合
return True if stack == [] else False
if __name__ == '__main__':
s = Solution()
pushV = [1, 2, 3, 4, 5]
popV = [4, 3, 5, 1, 2]
result = s.isPopOrder(pushV, popV)
print(result)
print('-----------')
popV1 = [4, 3, 5, 2, 1]
result = s.isPopOrder(pushV, popV1)
print(result)