题目
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。
假设压入栈的所有数字均不相等。
例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。
注意:若两个序列长度不等则视为并不是一个栈的压入、弹出序列。若两个序列都为空,则视为是一个栈的压入、弹出序列。
样例
输入:[1,2,3,4,5]
[4,5,3,2,1]
输出:true
解法:模拟
总体思路是借助辅助栈,按照压入顺序将数字压入辅助栈,看是否能按照弹出顺序将数字弹出
可以发现这样的规律:如果辅助栈栈顶数字和弹出顺序当前数字相同的话,那么一定是要弹出的,否则就继续向辅助栈中加入数字
模拟上述过程,直到所有数字都被压入过辅助栈,如果最后辅助栈为空的话,那么弹出序列就是合法的
时间复杂度O(n),空间复杂度O(1)
class Solution {
public:
bool isPopOrder(vector<int> pushV,vector<int> popV) {
if (pushV.size() != popV.size()) return false;
stack<int> s;
int i = 0;
for (auto x: pushV) {
s.push(x);
while (s.size() && s.top() == popV[i]) {
s.pop();
i++;
}
}
return s.empty();
}
};