题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
代码如下:
#include <stack>
class Solution {
public:
bool IsPopOrder(vector<int> pushV, vector<int> popV) {
stack<int> st;//存储部分入栈元素
vector<int>::iterator pushIt = pushV.begin(), popIt;
for (popIt = popV.begin(); popIt != popV.end(); ++popIt){
for (; pushIt != pushV.end(); ++pushIt){
if (!st.empty()){
int top = st.top();
if (top == (*popIt)){
st.pop();
break;
}
}
if ((*popIt) != (*pushIt))
st.push(*pushIt);
else{
++pushIt;
if (pushIt == pushV.end())//匹配了导致数组比较完则popIt指针加一。
popIt ++;
break;
}
}
if (pushIt == pushV.end()) break;
}
//入栈数组入栈和跟出栈数组比较完毕才执行跟栈顶元素比较的操作。
for (; popIt != popV.end(); ++popIt){
if (st.top() == (*popIt)){
st.pop();
}
else{
return false;
}
}
return true;
}
};