题目
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
思路
两个数组A,B,其中A为入栈序列,B为出栈序列。 借助一个辅助栈,先压入A中首元素,然后与B中首元素对比,如果相同则出栈,B的元素下标+1作为首元素;如果不相同则继续压入A中元素;如果B的元素下标等于B数组的size时,说明符合出栈序列,返回true;如果A中元素的下标等于A.size()时,说明不符合出栈顺序(1、此时出栈数组的首元素暂时无法从辅助栈出栈;2、此时出栈序列中首元素不存在入栈序列)
示例
入栈示例:
3 4 5 7
出栈序列一:
4 5 3 7
出栈序列二:
3 5 7 4
出栈序列三:
5 3 4 7
结果:
序列一: 1
序列二: 1
序列三: 0
代码
#include <iostream>
#include <stack>
#include <vector>
using namespace std;
class Solution{
public:
bool IsPopOder(vector<int> pushV, vector<int> popV){
auxiliary.push(pushV[0]);
int popPosit(0);//出栈序列的位置
for (int i = 1; i < pushV.size()+1;) {
if (auxiliary.empty()) { // 如果辅助栈为空,则把入栈序列首元素压入辅助栈
auxiliary.push(pushV[i]);
i++;
} else if (auxiliary.top() == popV[popPosit]){ //如果辅助栈栈顶等于出栈数组的首元素,将栈顶元素出栈
auxiliary.pop();
popPosit++;
// 如果出栈序列的位置等于size,则符合
if (popPosit == popV.size())
return true;
} else {
// 如果入栈的数组下标等于size则说明出栈序列不符合出栈顺序
if (i == pushV.size())
return false;
auxiliary.push(pushV[i]);
i++;
}
}
}
private:
stack<int> auxiliary;
};
int main(){
vector<int> pushV = {3,4,5,7};
vector<int> popVR = {4,5,3,7};
vector<int> popVE = {3,5,7,4};
vector<int> popVE2 = {5,3,4,7};
Solution re;
cout << "入栈示例:" << endl ;
for (int i = 0; i < pushV.size() ; ++i) {
cout << pushV[i] << " ";
}
cout << endl << "出栈序列一: " << endl;
for (int j = 0; j < popVR.size() ; ++j) {
cout << popVR[j] << " ";
}
cout << endl << "出栈序列二: " << endl;
for (int k = 0; k < popVE.size() ; ++k) {
cout << popVE[k] << " ";
}
cout << endl << "出栈序列三:" << endl;
for (int l = 0; l < popVE2.size() ; ++l) {
cout << popVE2[l] << " ";
}
cout << endl << "结果: " << endl;
cout << "序列一: " << re.IsPopOder(pushV,popVR) << endl;
cout << "序列二: " << re.IsPopOder(pushV,popVE) << endl;
cout << "序列三: " << re.IsPopOder(pushV,popVE2) << endl;
return 0;
}