剑指offer-栈的压入、弹出序列
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
# -*- coding:utf-8 -*- class Solution: def IsPopOrder(self, pushV, popV): # write code here stack = [] j = 0 for i in pushV: stack.append(i) while stack: if stack and stack[-1] == popV[j]: stack.pop() j += 1 else: break if stack: return False else: return True
思路和我们人脑计算的思路是一致的
- 建立一个辅助栈 stack = [];
- 将压入序列的第一个数压入辅助栈,判断辅助栈顶的元素是否和弹出序列的第一个数(当前数)相同,如果相同的话辅助栈弹出该相同数(说明可以弹出序列的第一个数是对的),然后弹出序列移到第二个数(当前数);
- 如果不同继续将压入序列的数入辅助栈,再次比较辅助栈顶元素是否和弹出序列当前数相同。
- 直到压入序列中的数全部入辅助栈。
- 如果辅助栈stack中还有元素,说明不是弹出序列,如果没有说明弹出序列正确。