题目
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
分析
建立一个辅助栈,按照输入栈序列的顺序将元素入栈,并按照输出栈序列的顺序依次将辅助栈的元素出栈。具体思路如下:
- 首先判断输入输出序列至少其中之一没有元素,若存在直接返回false;
- 遍历所有输入序列,若辅助栈为空或者辅助栈栈顶元素与当前指针对应的输出序列元素不等,就输入序列元素入栈,当辅助栈栈顶元素与当前指针对应的输出序列元素相等则将辅助栈元素出栈即可。
github链接:JZ21-栈的压入、弹出序列
C++代码
#include <iostream>
#include <stack>
#include <vector>
using namespace std;
class Solution {
public:
bool IsPopOrder(vector<int> pushV,vector<int> popV) {
if(pushV.size() == 0 || popV.size() == 0){
return false;
}
int start = 0;
stack<int> s;
int m = pushV.size();
for(int i = 0 ; i < m ; i++){
while(s.empty() || s.top() != popV[i]){
if(start >= m){
return false;
}
s.push(pushV[start++]);
}
s.pop();
}
return true;
}
};
int main()
{
int m,n;
while(cin>>m>>n){
vector<int> push(m),pop(n);
for(int i = 0 ; i < m ; i++){
cin>>push[i];
}
for(int i = 0 ; i < n ; i++){
cin>>pop[i];
}
Solution s;
cout<<s.IsPopOrder(push,pop);
}
return 0;
}